torch.backends

Creator

Created

2023 Aug 21 16:11

Editor

Edited

2023 Aug 21 16:11

Refs


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

torch backends

////////