Skip to content
Snippets Groups Projects
Select Git revision
  • pointer-regression
  • master default protected
2 results
Name Last commit Last update
hpvm/benchmarks/dnn_benchmarks/keras
hpvm/benchmarks/dnn_benchmarks/pytorch
hpvm/benchmarks/dnn_benchmarks/tensor-rt-src
hpvm/benchmarks/dnn_benchmarks/.gitignore
hpvm/benchmarks/dnn_benchmarks/README.rst
README.rst

DNN Benchmarks

Directory Organization

  • keras contains these DNNs implemented in Keras, and code for generating them down to HPVM-C (testing Keras frontend).
  • pytorch contains these DNNs in PyTorch and code for generating them down to HPVM-C (testing PyTorch/ONNX frontend).
    • ./dnn is a local package with these 10 DNNs implemented in PyTorch as examples. This package is not installed with HPVM.
    • ./test_frontend contains tests on inference accuracy of code generated by the PyTorch frontend.
    • ./test_{profiling|tuning} contains tests on performing profiling/tuning on frontend-generated binary.
  • tensor-rt-src contains these DNNs directly implemented in tensor_runtime functions. These are for reference purpose only and not actively used in the HPVM system or testing.