test: add note

Mayukhdeb · Mayukhdeb · commit 661f4fa2cbb3 · 2025-01-02T12:18:06.000-05:00
diff --git a/tests/test_loss_linear.py b/tests/test_loss_linear.py
@@ -7,7 +7,7 @@
 supported_dtypes = [
     torch.float32,
     torch.float16,
-    #torch.bfloat16,
+    # torch.bfloat16, ### tests fail on bfloat16
 ]
 
 # Define the fixture that provides the num_steps argument

Original file line number	Diff line number	Diff line change
`@@ -7,7 +7,7 @@`
`7`	`7`	`supported_dtypes = [`
`8`	`8`	`torch.float32,`
`9`	`9`	`torch.float16,`
`10`		`- #torch.bfloat16,`
	`10`	`+ # torch.bfloat16, ### tests fail on bfloat16`
`11`	`11`	`]`
`12`	`12`
`13`	`13`	`# Define the fixture that provides the num_steps argument`