Skip to content

Latest commit

 

History

History

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 

README.md

Shift/Add operators on TVM

Test TVM-based shift/add operators on computed shapes that appear in PVT

Build up TVM:

Please follow TVM documentation to compile. "set(USE_CUDA ON)" is necessary

Since we need to auto-tuning module, Python dependencies below is needed.

pip install tornado psutil 'xgboost>=1.1.0' cloudpickle

Shift op:

Please move to the corresponding folder first.

cd ./OPs_Speedups/Shift
  1. build op and tune:
python Ansor_tune.py
  1. build op and test latency
python latency_test.py

Add op:

Please move to the corresponding folder first.

cd ./OPs_Speedups/Add
  1. build op and tune:
python Ansor_tune.py
  1. build op and test latency
python latency_test.py