Tensor float 32

Created

Created

2021 May 13 9:52

Language

Language

Creator

Creator

Seonglae Cho

Editor

Editor

Seonglae Cho

Edited

Edited

2024 Mar 8 8:11

Refs

Refs


torch.backends.cuda.matmul.allow_tf32 = True  # allow tf32 on matmul
torch.backends.cudnn.allow_tf32 = True  # allow tf32 on cudnn

Recommendations

////////////