File tree Expand file tree Collapse file tree 2 files changed +17
-2
lines changed Expand file tree Collapse file tree 2 files changed +17
-2
lines changed Original file line number Diff line number Diff line change 8
8
9
9
# noinspection PyUnresolvedReferences
10
10
from mlir .extras .testing import MLIRContext , filecheck , mlir_ctx as ctx
11
- from util import llvm_bindings_not_installed
11
+ from util import llvm_bindings_not_installed , llvm_amdgcn_bindings_not_installed
12
12
13
13
# needed since the fix isn't defined here nor conftest.py
14
14
pytest .mark .usefixtures ("ctx" )
15
15
16
16
17
- @pytest .mark .skipif (llvm_bindings_not_installed (), reason = "llvm bindings not installed" )
17
+ @pytest .mark .skipif (
18
+ llvm_bindings_not_installed () or llvm_amdgcn_bindings_not_installed (),
19
+ reason = "llvm bindings not installed or llvm_amdgcn bindings not installed" ,
20
+ )
18
21
def test_call_instrinsic (ctx : MLIRContext ):
19
22
@func (emit = True )
20
23
def sum (a : T .i32 (), b : T .i32 (), c : T .f32 ()):
Original file line number Diff line number Diff line change @@ -38,6 +38,18 @@ def llvm_bindings_not_installed():
38
38
return True
39
39
40
40
41
+ def llvm_amdgcn_bindings_not_installed ():
42
+ try :
43
+ from mlir .extras .dialects .ext .llvm import amdgcn
44
+
45
+ # don't skip
46
+ return False
47
+
48
+ except ImportError :
49
+ # skip
50
+ return True
51
+
52
+
41
53
def hip_check (call_result ):
42
54
from hip import hip
43
55
You can’t perform that action at this time.
0 commit comments