-
- Downloads
[ROCM] MIOpen contrib for convolution kernels (#722)
* fist working miopen support * do FindFwdAlgo during build time * fix lint * update doc string * import topi after checking if rocm is enabled * add miopen namespace * fixed descriptor overwrite bug * add use_miopen option * fix lint * better miopen option handling * fix typo * fix options handling
Showing
- Makefile 1 addition, 0 deletionsMakefile
- make/config.mk 3 additions, 0 deletionsmake/config.mk
- make/contrib/miopen.mk 8 additions, 0 deletionsmake/contrib/miopen.mk
- python/tvm/contrib/miopen.py 102 additions, 0 deletionspython/tvm/contrib/miopen.py
- python/tvm/target.py 8 additions, 3 deletionspython/tvm/target.py
- src/contrib/miopen/conv_forward.cc 221 additions, 0 deletionssrc/contrib/miopen/conv_forward.cc
- src/contrib/miopen/miopen_utils.cc 78 additions, 0 deletionssrc/contrib/miopen/miopen_utils.cc
- src/contrib/miopen/miopen_utils.h 59 additions, 0 deletionssrc/contrib/miopen/miopen_utils.h
- tests/python/contrib/test_miopen.py 64 additions, 0 deletionstests/python/contrib/test_miopen.py
- tests/scripts/task_python_unittest.sh 1 addition, 1 deletiontests/scripts/task_python_unittest.sh
Loading
Please register or sign in to comment