|
|
@@ -4,11 +4,6 @@ from tinygrad.helpers import getenv
|
|
|
|
|
|
|
|
|
def test_bf16():
|
|
|
- weights = {
|
|
|
- "weight1": Tensor.randn(10, 10, dtype=dtypes.bfloat16),
|
|
|
- "weight2": Tensor.randn(20, 20, dtype=dtypes.float32)
|
|
|
- }
|
|
|
-
|
|
|
result = Tensor.randn(10, 10, dtype=dtypes.bfloat16) + Tensor.randn(10, 10, dtype=dtypes.bfloat16)
|
|
|
result.realize()
|
|
|
print(result)
|