Skip to content

Reference implementation of Deep Neural Network primitives using LIBXSMM's Tensor Processing Primitives (TPP)

License

Notifications You must be signed in to change notification settings

libxsmm/libxsmm-dnn

Repository files navigation

LIBXSMM-DNN

BSD 3-Clause License Correctness Performance Read the Docs

LIBXSMM-DNN is a reference implementation of Deep Neural Network primitives using LIBXSMM's Tensor Processing Primitives (TPP).

To checkout the source code:

git pull --recurse-submodules
git submodule update --init --recursive

To build from source:

make -j $(nproc)

NOTE: To rely on a specific version of LIBXSMM (Git submodule not required), set LIBXSMMROOT environment/make variable and rely on a separately built LIBXSMM.

To mimic the performance of ResNet-50 inference:

tests/conv/run_convs.sh 1 1000 -1 f32 F 0 0 64 64 1 | tee output.log

NOTE: To mimic performance, FLOPS are shown and the time spent for convolutional layers is measured. This is different from running inference end-to-end using a real framework such as PyTorch).

About

Reference implementation of Deep Neural Network primitives using LIBXSMM's Tensor Processing Primitives (TPP)

Resources

License

Security policy

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 6