Skip to content

Commit ea19842

Browse files
committed
exclude TF32 tests for Julia <= 1.7
1 parent 181691b commit ea19842

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

test/device/intrinsics/wmma.jl

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,11 @@ end
4242
continue
4343
end
4444

45+
if mnk == (16,16,8) && VERSION <= v"1.7"
46+
# TensorFlow32 tests require at least Julia 1.8
47+
continue
48+
end
49+
4550
shape = CUDA.WMMA.get_hl_shape(mnk[1], mnk[2], mnk[3])
4651

4752
# Type-dependent variables
@@ -103,6 +108,11 @@ end
103108
continue
104109
end
105110

111+
if mnk == (16,16,8) && VERSION <= v"1.7"
112+
# TensorFlow32 tests require at least Julia 1.8
113+
continue
114+
end
115+
106116
shape = CUDA.WMMA.get_hl_shape(mnk[1], mnk[2], mnk[3])
107117

108118
# Type-dependent variables
@@ -157,6 +167,11 @@ end
157167
d_elem_type in ops[4],
158168
c_elem_type in ops[3]
159169

170+
if mnk == (16,16,8) && VERSION <= v"1.7"
171+
# TensorFlow32 tests require at least Julia 1.8
172+
continue
173+
end
174+
160175
# Type-dependent variables
161176
d_ty = CUDA.WMMA.map_ptx_to_jl_array[d_elem_type]
162177
c_ty = CUDA.WMMA.map_ptx_to_jl_array[c_elem_type]

0 commit comments

Comments
 (0)