migraphx-benchmark / AMDMIGraphX

AMD's graph optimization engine.
https://rocmsoftwareplatform.github.io/AMDMIGraphX/doc/html/
MIT License
0 stars 1 forks source link

Common API for a simple TensorRT script #187

Closed mirza-halilcevic closed 1 month ago

mirza-halilcevic commented 2 months ago

Implement an API that allows us to run a simple TensorRT script using MIGraphX with minimal effort. The simple script should parse an ONNX file, invoke the builder, produce the engine, and run inference. Verify that MIGraphX produces the correct results.

Consider using sampleOnnxMNIST from the TensorRT repo as a starting point.