Skip to content

Commit 6b321f2

Browse files
committed
Fixed the HIP register observer test
1 parent 8989a07 commit 6b321f2

File tree

1 file changed

+15
-8
lines changed

1 file changed

+15
-8
lines changed

test/test_observers.py

Lines changed: 15 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,21 @@
1+
from pytest import raises
2+
13
import kernel_tuner
24
from kernel_tuner.observers.nvml import NVMLObserver
3-
from kernel_tuner.observers.register import RegisterObserver
45
from kernel_tuner.observers.observer import BenchmarkObserver
6+
from kernel_tuner.observers.register import RegisterObserver
57

6-
from .context import skip_if_no_pycuda, skip_if_no_pynvml, skip_if_no_cupy, skip_if_no_cuda, skip_if_no_opencl, skip_if_no_pyhip
8+
from .context import (
9+
skip_if_no_cuda,
10+
skip_if_no_cupy,
11+
skip_if_no_opencl,
12+
skip_if_no_pycuda,
13+
skip_if_no_pyhip,
14+
skip_if_no_pynvml,
15+
)
16+
from .test_hip_functions import env as env_hip # noqa: F401
17+
from .test_opencl_functions import env as env_opencl # noqa: F401
718
from .test_runners import env # noqa: F401
8-
from .test_opencl_functions import env as env_opencl # noqa: F401
9-
from .test_hip_functions import env as env_hip # noqa: F401
10-
11-
from pytest import raises
1219

1320

1421
@skip_if_no_pycuda
@@ -62,8 +69,8 @@ def test_register_observer_opencl(env_opencl):
6269
assert "OpenCL" in str(err.value)
6370

6471
@skip_if_no_pyhip
65-
def test_register_observer_hip(env_opencl):
72+
def test_register_observer_hip(env_hip):
6673
with raises(NotImplementedError) as err:
67-
kernel_tuner.tune_kernel(*env_opencl, observers=[RegisterObserver()], lang='HIP')
74+
kernel_tuner.tune_kernel(*env_hip, observers=[RegisterObserver()], lang='HIP')
6875
assert err.errisinstance(NotImplementedError)
6976
assert "Hip" in str(err.value)

0 commit comments

Comments
 (0)