[Script] Add mising no_grad context

Author: YWHyuk

experiments/BERT.py

@@ -8,7 +8,7 @@
 import torch
 from Simulator.simulator import TOGSimulator
 
-config = os.environ.get('TOGSIM_CONFIG', f'{base_path}/configs/systolic_ws_128x128_c1_simple_noc_tpuv3.yml')
+config = os.environ.get('TOGSIM_CONFIG', f'{base_path}/configs/systolic_ws_128x128_c1_simple_noc_tpuv3_timing_only.yml')
 os.environ['TOGSIM_CONFIG'] = config
 
 # Try Fusion EncoderBlock first, fall back to standard test_transformer
@@ -36,7 +36,7 @@
     model_input = torch.randn(args.input_size, hidden_dim).to(device=device)
     opt_fn = torch.compile(dynamic=False)(model)
 
-    with TOGSimulator(config_path=config):
+    with TOGSimulator(config_path=config), torch.no_grad():
         torch.npu.launch_model(opt_fn, model_input, stream_index=0, timestamp=0)
         torch.npu.synchronize()
     print(f"BERT-{args.size} Simulation Done")

experiments/resnet18.py

@@ -22,7 +22,7 @@
     opt_fn = torch.compile(dynamic=False)(model)
     model_input = torch.randn(args.batch, 3, 224, 224).to(device=device)
 
-    with TOGSimulator(config_path=config):
+    with TOGSimulator(config_path=config), torch.no_grad():
         torch.npu.launch_model(opt_fn, model_input, stream_index=0, timestamp=0)
         torch.npu.synchronize()
     print("ResNet18 Simulation Done")

experiments/resnet50.py

@@ -22,7 +22,7 @@
     opt_fn = torch.compile(dynamic=False)(model)
     model_input = torch.randn(args.batch, 3, 224, 224).to(device=device)
 
-    with TOGSimulator(config_path=config):
+    with TOGSimulator(config_path=config), torch.no_grad():
         torch.npu.launch_model(opt_fn, model_input, stream_index=0, timestamp=0)
         torch.npu.synchronize()
     print("ResNet50 Simulation Done")