update torch cuda device_type to dlpack kDLROCM=10

822caeb6 · jayzlee147 · 0ef85f28 · 822caeb6
Commit 822caeb6 authored May 31, 2023 by jayzlee147
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

python/dgl/_ffi/runtime_ctypes.py python/dgl/_ffi/runtime_ctypes.py +1 -1

No files found.
--- a/python/dgl/_ffi/runtime_ctypes.py
+++ b/python/dgl/_ffi/runtime_ctypes.py
@@ -121,7 +121,7 @@ class DGLContext(ctypes.Structure):
        'stackvm': 1,
        'cpu': 1,
        'gpu': 2,
-        'cuda': 2,
+        'cuda': 10,
        'nvptx': 2,
        'cl': 4,
        'opencl': 4,