diff --git a/scripts/skiplist/default/triton_kernels.txt b/scripts/skiplist/default/triton_kernels.txt index f3a4d1078e..8f96074914 100644 --- a/scripts/skiplist/default/triton_kernels.txt +++ b/scripts/skiplist/default/triton_kernels.txt @@ -71,3 +71,4 @@ tests/test_matmul.py::test_op[False-True-True-True-False-None-16-1000-704-800-ra tests/test_matmul.py::test_op[False-True-True-True-False-None-16-1000-704-800-ragged-mxfloat8_e4m3fn-mxfloat4_e2m1-8-2-9-True-None-False-False-False] tests/test_matmul.py::test_op[False-True-True-True-False-None-16-16-16-1000-batched-float16-float16-5-1-None-False-None-False-False-False] tests/test_matmul.py::test_op[False-True-True-True-False-None-16-16-16-1000-batched-float8_e5m2-float8_e5m2-5-1-None-False-None-False-False-False] +tests/test_matmul.py::test_op[False-False-False-False-False-None-128-1000-700-700-ragged-float16-float16-8-2-1-False-None-False-False-False] diff --git a/third_party/intel/lib/TritonIntelGPUToLLVM/LoadStoreOpToLLVM.cpp b/third_party/intel/lib/TritonIntelGPUToLLVM/LoadStoreOpToLLVM.cpp index aaa8dec97a..6c1dcd9d22 100644 --- a/third_party/intel/lib/TritonIntelGPUToLLVM/LoadStoreOpToLLVM.cpp +++ b/third_party/intel/lib/TritonIntelGPUToLLVM/LoadStoreOpToLLVM.cpp @@ -2956,10 +2956,7 @@ struct LoadOpConversion : public ConvertOpToLLVMPattern, Value ret; // Create a predicated load operation. if (pred) { - std::optional enablePredicated = - mlir::triton::tools::isEnvValueBool( - mlir::triton::tools::getStrEnv("TRITON_INTEL_PREDICATED")); - if (!enablePredicated.has_value() || enablePredicated.value()) + if (triton::tools::getBoolEnv("TRITON_INTEL_PREDICATED")) ret = rewriter.create( loc, retTy, addrElem, b.i64_val(alignment), pred, other_); else {