Disable TF32 usage by default.

RaulPPelaez · RaulPPelaez · commit 0f49ca1d7fe7 · 2024-05-21T15:26:46.000+02:00
diff --git a/torchmdnet/models/tensornet.py b/torchmdnet/models/tensornet.py
@@ -13,8 +13,6 @@
 )
 
 __all__ = ["TensorNet"]
-torch.set_float32_matmul_precision("high")
-torch.backends.cuda.matmul.allow_tf32 = True
 
 
 def vector_to_skewtensor(vector):

Original file line number	Diff line number	Diff line change
`@@ -13,8 +13,6 @@`
`13`	`13`	`)`
`14`	`14`
`15`	`15`	`__all__ = ["TensorNet"]`
`16`		`-torch.set_float32_matmul_precision("high")`
`17`		`-torch.backends.cuda.matmul.allow_tf32 = True`
`18`	`16`
`19`	`17`
`20`	`18`	`def vector_to_skewtensor(vector):`