[gtfn] Fix to detect CUPY gpu runtime

edopao · edopao · commit b6ed1cf855dd · 2024-02-08T13:27:27.000+01:00
diff --git a/src/gt4py/next/allocators.py b/src/gt4py/next/allocators.py
@@ -44,7 +44,7 @@
 CUPY_DEVICE: Final[Literal[None, core_defs.DeviceType.CUDA, core_defs.DeviceType.ROCM]] = (
     None
     if not cp
-    else (core_defs.DeviceType.ROCM if cp.cuda.get_hipcc_path() else core_defs.DeviceType.CUDA)
+    else (core_defs.DeviceType.ROCM if cp.cuda.runtime.is_hip else core_defs.DeviceType.CUDA)
 )
 
 

Original file line number	Diff line number	Diff line change
`@@ -44,7 +44,7 @@`
`44`	`44`	`CUPY_DEVICE: Final[Literal[None, core_defs.DeviceType.CUDA, core_defs.DeviceType.ROCM]] = (`
`45`	`45`	`None`
`46`	`46`	`if not cp`
`47`		`- else (core_defs.DeviceType.ROCM if cp.cuda.get_hipcc_path() else core_defs.DeviceType.CUDA)`
	`47`	`+ else (core_defs.DeviceType.ROCM if cp.cuda.runtime.is_hip else core_defs.DeviceType.CUDA)`
`48`	`48`	`)`
`49`	`49`
`50`	`50`