Skip to content

Commit

Permalink
Remove old LLVM bug workaround. (#141)
Browse files Browse the repository at this point in the history
Signed-off-by: Ilya Enkovich <[email protected]>
  • Loading branch information
ienkovich authored Sep 6, 2024
1 parent 6a333f1 commit 0b65730
Showing 1 changed file with 0 additions and 6 deletions.
6 changes: 0 additions & 6 deletions python/test/unit/language/test_core.py
Original file line number Diff line number Diff line change
Expand Up @@ -2236,12 +2236,6 @@ def kernel(X, Z, BLOCK: tl.constexpr):
def test_reduce(op, dtype_str, shape, axis, keep_dims, num_ctas, device):
check_type_supported(dtype_str, device) # bfloat16 on cc < 80 will not be tested

# fpext fp16->fp32 is broken in LLVM for large vectors:
# https://github.com/llvm/llvm-project/issues/95278
# TODO: remove the change after the bug is fixed.
if is_cpu() and dtype_str == "float16":
shape = (min(shape[0], 512), min(shape[1], 512))

@triton.jit
def kernel(X, Z, BLOCK_M: tl.constexpr, BLOCK_N: tl.constexpr, BLOCK_K: tl.constexpr, IS_3D: tl.constexpr,
AXIS: tl.constexpr, KEEP_DIMS: tl.constexpr, USE_I1: tl.constexpr):
Expand Down

0 comments on commit 0b65730

Please sign in to comment.