frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add missing caffe2_hip extension in setup.py

Browse Source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/15189

Reviewed By: orionr

Differential Revision: D13457644

Pulled By: bddppq

fbshipit-source-id: c2363e9b8fd21709b62777e5b2199f01ec1c65f8
This commit is contained in:
Junjie Bai
2018-12-13 15:57:20 -08:00
committed by Facebook Github Bot
parent de0784510d
commit bdfff2f8c2
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
setup.py
View File

@ -908,6 +908,12 @@ if USE_CUDA:
name=str('caffe2.python.caffe2_pybind11_state_gpu'), name=str('caffe2.python.caffe2_pybind11_state_gpu'),
sources=[]), sources=[]),
) )
if USE_ROCM:
extensions.append(
Extension(
name=str('caffe2.python.caffe2_pybind11_state_hip'),
sources=[]),
)
cmdclass = { cmdclass = {
'create_version_file': create_version_file, 'create_version_file': create_version_file,