Skip to content
Snippets Groups Projects

ApproxHPVM Tensor Runtime

Getting Started

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

  • tensor_runtime/include/tensor_runtime.h declares all the functions available in the runtime.

    TODO: the tensor runtime is generally under-documented at the time. More documentation will be added in the first public release.

  • 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_sources1: