debugging

Signed-off-by: Daniel Korzekwa <[email protected]>

Author: danielkorzekwa

tests/gpu/torch/prune/plugins/test_mcore_gpt_minitron_pruning.py

@@ -96,6 +96,9 @@ def _get_model(initialize_megatron=True):
     # (get_mcore_gpt_model calls initialize_for_megatron which sets seed=1234)
     torch.manual_seed(1234)
     torch.cuda.manual_seed_all(1234)
+    # Enable deterministic behavior for cuDNN
+    torch.backends.cudnn.deterministic = True
+    torch.backends.cudnn.benchmark = False
 
     sd = model.state_dict()