Skip to content

attafosu/tpp-pytorch-extension

 
 

Repository files navigation

Intel(R) Tensor Processing Primitives extension for PyTorch*

Copyright (c) Intel corp.

Pre-requisite

gcc v8.3.0 or higher

Installation

Setup conda environment using utils/setup_conda.sh

# Create new conda env  # It creates an env.sh script for activating conda env $bash utils/setup_conda.sh [-p <conda_install_path>]

Install the extension:

# Source the env.sh and install the extension $source env.sh $git submodule update --init $python setup.py install 

For multi-node runs:

(Optional) install torch_ccl module:

$bash utils/install_torch_ccl.sh

Examples

BERT

About

Intel® Tensor Processing Primitives extension for Pytorch*

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C++ 70.9%
  • Python 27.3%
  • Shell 1.6%
  • Other 0.2%