Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "b02e2113ff4625100a4412abd1ae0392ee415364"
Commit 9379df85 authored by justheuristic's avatar justheuristic
Browse files

check dtypes first

parent 140cdbe8
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_autograd.py
View file @ 9379df85
...@@ -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 == torch.dtype assert out_bnb.dtype == out_torch.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