github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-05 03:01:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

[ROCM] fix device_type name (#2061)

Browse Source 
Rename "rocm" -> "hip", to comply with other uses in compiler.py.
This commit is contained in:
Alexander Efimov
2023-08-09 04:11:09 +03:00
committed by GitHub
parent 6dee55c912
commit 1c45836d5d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/triton/compiler/compiler.py
View File

@@ -691,7 +691,7 @@ class CompiledKernel:
def runner(*args, stream=None):
args_expand = self.assemble_tensormap_to_arg(args, self.constants)
if stream is None:
if self.device_type in ["cuda", "rocm"]:
if self.device_type in ["cuda", "hip"]:
stream = get_cuda_stream()
else:
stream = get_backend(self.device_type).get_stream(None)