We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a90f114 commit 78613adCopy full SHA for 78613ad
1 file changed
experiments/gemm.py
@@ -31,7 +31,7 @@ def custom_matmul(a, b):
31
import os
32
import sys
33
base_dir = os.environ.get('TORCHSIM_DIR', default='/workspace/PyTorchSim')
34
- config = os.environ.get('TORCHSIM_CONFIG', default=f'{base_dir}/configs/systolic_ws_128x128_c2_simple_noc_tpuv4.yml)
+ config = os.environ.get('TORCHSIM_CONFIG', default=f'{base_dir}/configs/systolic_ws_128x128_c2_simple_noc_tpuv4.yml')
35
config_prefix = config.split('/')[-1].split('.')[0][9:] # extract config name from config path
36
sys.path.append(base_dir)
37
args = argparse.ArgumentParser()
0 commit comments