From c7b564bed65615c77f1065dd4821b5a79d2244b3 Mon Sep 17 00:00:00 2001
From: zy69guqi <richard.angersbach@fau.de>
Date: Wed, 19 Feb 2025 19:04:57 +0100
Subject: [PATCH] Fix CUDARuntimeError import
---
tests/kernelcreation/test_reduction.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tests/kernelcreation/test_reduction.py b/tests/kernelcreation/test_reduction.py
index 992c328d7..c3775964b 100644
--- a/tests/kernelcreation/test_reduction.py
+++ b/tests/kernelcreation/test_reduction.py
@@ -2,7 +2,6 @@ import pytest
import numpy as np
import pystencils as ps
-from cupy_backends.cuda.api.runtime import CUDARuntimeError
from pystencils.sympyextensions import reduction_assignment_from_str
INIT_W = 5
@@ -51,6 +50,7 @@ def test_reduction_cpu(instruction_set, dtype, op):
def test_reduction_gpu(dtype, op):
try:
import cupy as cp
+ from cupy_backends.cuda.api.runtime import CUDARuntimeError
device_count = range(cp.cuda.runtime.getDeviceCount())
print(f"Found {device_count} GPUs")
--
GitLab