pytorch/test/cpp_extensions
ashish 616beb1412 [ROCm] Added support for pytorch extensions to use HIP (#32669)
Summary:
This pull request has changes for:
1. Enabling a torch module with HIP code to be compiled by cpp_extensions.py
2. Fixes for hipify module to be able to be used by a torch extension

cc: ezyang iotamudelta jeffdaily
Pull Request resolved: https://github.com/pytorch/pytorch/pull/32669

Differential Revision: D20033893

Pulled By: zou3519

fbshipit-source-id: fd6ddc8cdcd3930f41008636bb2bc9dd26cdb008
2020-02-21 12:10:02 -08:00
..
no_python_abi_suffix_test
torch_test_cpp_extension
complex_registration_extension.cpp Rename TensorTypeId to DispatchKey (#32154) 2020-01-15 11:16:08 -08:00
cpp_frontend_extension.cpp
cuda_extension.cpp Deprecate tensor.type() (#30281) 2019-12-05 10:55:34 -08:00
cuda_extension.cu
cuda_extension_kernel.cu
cuda_extension_kernel2.cu
cudnn_extension.cpp
doubler.h
extension.cpp
jit_extension.cpp
jit_extension2.cpp
msnpu_extension.cpp Rename TensorTypeId to DispatchKey (#32154) 2020-01-15 11:16:08 -08:00
setup.py [ROCm] Added support for pytorch extensions to use HIP (#32669) 2020-02-21 12:10:02 -08:00