HPVM Tensor Runtime

Dependencies

• CUDA >= 9.1

• cuDNN >= 7

Building Tensor Runtime

Tensor Runtime and the DNN sources using the Tensor runtime are built with the unified HPVM build system. These can also be separately built. HPVM Tensor Runtime can be built as:

make -j ${NUM_THREADS} tensor_runtime

The tensor runtime is built as a static library under build/lib/liibtensor_runtime.a

Tensor Runtime APIs

• For examples of using tensor_runtime functions, see dnn_sources/src/alexnet_cifar10.cc.
  • Also, try running build/alexnet_cifar10 which is compiled from that file and runnable out of the box.

Directory Structure

• tensor_runtime:

• dnn_sources: