pytorch/test/custom_operator
jpvillam 6b21a33795 [ROCM] Enable custom tests on rocm
Enables custom test building and testing on ROCm.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/75473
Approved by: https://github.com/jeffdaily, https://github.com/janeyx99
2022-04-26 14:22:14 +00:00
..
CMakeLists.txt [ROCM] Enable custom tests on rocm 2022-04-26 14:22:14 +00:00
model.py Allow building libraries with setuptools that dont have abi suffix (#14130) 2018-11-27 17:35:53 -08:00
op.cpp use irange for loops 5 (#66744) 2021-10-18 21:59:50 -07:00
op.h Test cases for custom ops with autograd (#31003) 2019-12-15 22:37:24 -08:00
test_custom_classes.py Set test owners for tests with unknown owners (#67552) 2021-10-29 12:42:01 -07:00
test_custom_ops.cpp use irange for loops 5 (#66744) 2021-10-18 21:59:50 -07:00
test_custom_ops.py Set test owners for tests with unknown owners (#67552) 2021-10-29 12:42:01 -07:00