summarylogtreecommitdiffstats
path: root/pytorch-rocm-jit.patch
blob: 3afb3e43edbf6486c459c2f7fd459fd64f1269f9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
--- pytorch-rocm/torch/csrc/jit/ir/ir.cpp.bak	2023-07-08 14:23:12.514918311 +0200
+++ pytorch-rocm/torch/csrc/jit/ir/ir.cpp	2023-07-08 14:23:57.239106350 +0200
@@ -1186,10 +1186,10 @@
     case prim::rpc_remote: // It represents RPC message sent.
     case aten::wait: // It can represent RPC message received.
 #if !defined(USE_ROCM)
-    case cuda::set_stream:
-    case cuda::_set_device:
-    case cuda::_current_device:
-    case cuda::synchronize:
+    case c10::cuda::set_stream:
+    case c10::cuda::_set_device:
+    case c10::cuda::_current_device:
+    case c10::cuda::synchronize:
 #endif
     case prim::Enter:
     case prim::Exit: