Commit e29c5f5c authored by justheuristic's avatar justheuristic
Browse files

clearer assertions

parent 9379df85
......@@ -367,7 +367,7 @@ def test_matmullt(
if has_bias:
out_torch += bias
assert out_bnb.dtype == out_torch.dtype
assert out_bnb.dtype == A.dtype, f"bnb matmullt received {A.dtype} but returned {out_bnb.dtype}"
n = out_bnb.numel()
err = torch.abs(out_bnb - out_torch).mean().item()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment