From: Junjie Bai <bai@in.tum.de>
Date: Thu, 13 Dec 2018 23:57:20 +0000 (-0800)
Subject: Add missing caffe2_hip extension in setup.py
X-Git-Tag: accepted/tizen/6.5/unified/20211028.231830~2256
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=bdfff2f8c28faa71e8323ee5cb8a4957679e581c;p=platform%2Fupstream%2Fpytorch.git

Add missing caffe2_hip extension in setup.py

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

Reviewed By: orionr

Differential Revision: D13457644

Pulled By: bddppq

fbshipit-source-id: c2363e9b8fd21709b62777e5b2199f01ec1c65f8
---

diff --git a/setup.py b/setup.py
index 6f8323d..79ea3b7 100644
--- a/setup.py
+++ b/setup.py
@@ -908,6 +908,12 @@ if USE_CUDA:
             name=str('caffe2.python.caffe2_pybind11_state_gpu'),
             sources=[]),
     )
+if USE_ROCM:
+    extensions.append(
+        Extension(
+            name=str('caffe2.python.caffe2_pybind11_state_hip'),
+            sources=[]),
+    )
 
 cmdclass = {
     'create_version_file': create_version_file,