Skip to content

Commit e9c51be

Browse files
committed
Run pytest in python/examples and xfail unsupported tests
Also update Triton to 2efb067bfc0f9acabcd8b4ffe7c55ad248dfb282.
1 parent fe11a2d commit e9c51be

File tree

3 files changed

+26
-3
lines changed

3 files changed

+26
-3
lines changed

backend/compiler.py

+3
Original file line numberDiff line numberDiff line change
@@ -137,6 +137,9 @@ class CPUOptions:
137137
extern_libs = None
138138
cluster_dims: tuple = (1, 1, 1)
139139
shared: bool = False
140+
# Disable FP8 here since this is a sample CPU backend.
141+
# Target specific backends can eanble it with supported types.
142+
supported_fp8_dtypes: Tuple[str] = ()
140143
allow_fp8e4nv: bool = False
141144
allowed_dot_input_precisions: Tuple[str] = ("ieee", )
142145
sanitize_overflow: bool = True

python/examples/conftest.py

+22-2
Original file line numberDiff line numberDiff line change
@@ -71,12 +71,32 @@ def device(request):
7171
"test_if",
7272
"test_if_call",
7373
"test_convert2d",
74-
"test_convertmma2mma",
7574
"test_dot_max_num_imprecise_acc",
7675
"test_propagate_nan",
7776
"test_clamp_symmetric",
7877
"test_temp_var_in_loop",
79-
"test_math_extern"
78+
"test_math_extern",
79+
# attribute 'launch_cooperative_grid' not supported
80+
"test_load_scope_sem_coop_grid_cta_one",
81+
# fp8 support on CPUs is unclear
82+
"test_scaled_dot",
83+
# triton-shared-opt failures:
84+
# PtrAnalysis: encountered addptr operand produced by an unsupported operation
85+
"test_chained_reductions",
86+
# failed to legalize unresolved materialization
87+
"test_constexpr_if_return",
88+
"test_unroll_attr",
89+
# Dialect `ub' not found for custom op 'ub.poison'
90+
"test_poison_return",
91+
# tt.gather not supported yet
92+
"test_gather",
93+
"test_gather_warp_shuffle",
94+
# device 'cpu' does not have 'index
95+
"test_zero_strided_tensors",
96+
# hard-coded with 'ttg' attributes
97+
"test_convert_mma2mma",
98+
"test_local_load_store",
99+
"test_local_load_store_mma"
80100
}
81101

82102
# probably different version of MLIR on the nightly build machine is complaining

0 commit comments

Comments
 (0)