search

buddy-compiler / buddy-benchmark

Benchmark Framework for Buddy Projects
Apache License 2.0
44 stars 33 forks source link

[DL] Add vision benchmarks for IREE, TensorRT and PyTorch. #114

Closed SForeKeeper closed 1 week ago

SForeKeeper commented 7 months ago

This PR adds support for vision inference using IREE, TensorRT and PyTorch.

For IREE, torch-mlir is used for emitting MLIR code. For TensorRT, a builtin engine builder is used. For PyTorch, torch.compile in full-graph mode is used for maximizing performance.