Skip to content

Commit

Permalink
Add testcase for accumulating in fp32, but storing as bf16 (triton-la…
Browse files Browse the repository at this point in the history
…ng#1305)

No further changes were needed after triton-lang#1282. Fixes triton-lang#1264.

Signed-off-by: Julian Oppermann <julian.oppermann@codeplay.com>
  • Loading branch information
jopperm authored Jun 10, 2024
1 parent f57090e commit 2c589d6
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions python/tutorials/09-experimental-block-pointer.py
Original file line number Diff line number Diff line change
Expand Up @@ -218,6 +218,7 @@ def matmul(a, b, accum_dtype, res_dtype):
(torch.float16, torch.float32, torch.float16),
(torch.float16, torch.float32, torch.float32),
(torch.bfloat16, torch.float32, torch.float32),
(torch.bfloat16, torch.float32, torch.bfloat16),
(torch.float32, torch.float32, torch.float32),
(torch.int8, torch.int32, torch.int32)]:
if dtype.is_floating_point:
Expand Down

0 comments on commit 2c589d6

Please sign in to comment.