Skip to content
Open
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion examples/developer_mode/gelu_mul_developer.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,6 +73,6 @@ def main(A: T.Tensor((M, N), dtype), B: T.Tensor((M, N // 2), dtype)):
a1, a2 = torch.split(a, N // 2, dim=1)
ref_b = gelu(a1) * a2
torch.testing.assert_close(b.cpu(), ref_b.cpu(), rtol=1e-2, atol=1e-2)
print("Test passed!")
print("Test passed")

print("Kernel Output Match!")
Loading