[FRONTEND] provide device kwargs && fix fstring error for py<3.8 (#515)

Co-authored-by: Philippe Tillet <phil@openai.com>
This commit is contained in:
Jiabao Lei
2022-05-15 07:21:46 +08:00
committed by GitHub
parent d35617bea1
commit abea3dc2c6
2 changed files with 3 additions and 3 deletions

View File

@@ -644,7 +644,7 @@ def test_f16_to_f8_rounding():
)
assert torch.all(
torch.logical_not(mismatch)
), f"{f16_input[mismatch]=} {f16_output[mismatch]=} {abs_error[mismatch]=} {min_error[mismatch]=}"
), f"f16_input[mismatch]={f16_input[mismatch]} f16_output[mismatch]={f16_output[mismatch]} abs_error[mismatch]={abs_error[mismatch]} min_error[mismatch]={min_error[mismatch]}"
# ---------------