Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "97885cb1c983525dd14b82f38760221cd4ea7a17"
Commit e29c5f5c authored by justheuristic's avatar justheuristic
Browse files

clearer assertions

parent 9379df85
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_autograd.py
View file @ e29c5f5c
...@@ -367,7 +367,7 @@ def test_matmullt( ...@@ -367,7 +367,7 @@ def test_matmullt(
if has_bias: if has_bias:
out_torch += 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() n = out_bnb.numel()
err = torch.abs(out_bnb - out_torch).mean().item() 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