Test TVM-based shift/add operators on computed shapes that appear in PVT
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' cloudpicklePlease move to the corresponding folder first.
cd ./OPs_Speedups/Shift- build op and tune:
python Ansor_tune.py- build op and test latency
python latency_test.pyPlease move to the corresponding folder first.
cd ./OPs_Speedups/Add- build op and tune:
python Ansor_tune.py- build op and test latency
python latency_test.py