Skip to content

Commit b2725c2

Browse files
committed
remove debugging statements
1 parent 4dc8e86 commit b2725c2

1 file changed

Lines changed: 0 additions & 6 deletions

File tree

algoperf/pytorch_utils.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -21,12 +21,6 @@
2121

2222
def pytorch_setup() -> Tuple[bool, int, torch.device, int]:
2323
torch.set_float32_matmul_precision('high')
24-
# PyTorch set TF32
25-
# torch.backends.fp32_precision = "ieee"
26-
# torch.backends.cuda.matmul.fp32_precision = "tf32"
27-
# torch.backends.cudnn.fp32_precision = "ieee"
28-
# torch.backends.cudnn.conv.fp32_precision = "tf32"
29-
# torch.backends.cudnn.rnn.fp32_precision = "tf32"
3024

3125
use_pytorch_ddp = 'LOCAL_RANK' in os.environ
3226
rank = int(os.environ['LOCAL_RANK']) if use_pytorch_ddp else 0

0 commit comments

Comments
 (0)