drbh commited on
Commit
77e4af1
·
1 Parent(s): 7a0d4b3

fix: prefer hub kernel build

Browse files
Files changed (3) hide show
  1. compare_example.py +1 -1
  2. gpt_oss_backward.py +1 -1
  3. gpt_oss_match.py +1 -1
compare_example.py CHANGED
@@ -22,7 +22,7 @@ torch.backends.cudnn.benchmark = False
22
 
23
  np.set_printoptions(precision=4)
24
 
25
- load_method = 2 # 1: sym, 2: local, 3: hf
26
 
27
  if load_method == 1:
28
  sys.path.insert(0, "./torch-ext")
 
22
 
23
  np.set_printoptions(precision=4)
24
 
25
+ load_method = 3 # 1: sym, 2: local, 3: hf
26
 
27
  if load_method == 1:
28
  sys.path.insert(0, "./torch-ext")
gpt_oss_backward.py CHANGED
@@ -11,7 +11,7 @@ import time
11
  from kernels import get_kernel, get_local_kernel
12
  from pathlib import Path
13
 
14
- load_method = 2 # 1: sym, 2: local, 3: hf
15
 
16
  if load_method == 1:
17
  sys.path.insert(0, "./torch-ext")
 
11
  from kernels import get_kernel, get_local_kernel
12
  from pathlib import Path
13
 
14
+ load_method = 3 # 1: sym, 2: local, 3: hf
15
 
16
  if load_method == 1:
17
  sys.path.insert(0, "./torch-ext")
gpt_oss_match.py CHANGED
@@ -11,7 +11,7 @@ import time
11
  from kernels import get_kernel, get_local_kernel
12
  from pathlib import Path
13
 
14
- load_method = 2 # 1: sym, 2: local, 3: hf
15
 
16
  if load_method == 1:
17
  sys.path.insert(0, "./torch-ext")
 
11
  from kernels import get_kernel, get_local_kernel
12
  from pathlib import Path
13
 
14
+ load_method = 3 # 1: sym, 2: local, 3: hf
15
 
16
  if load_method == 1:
17
  sys.path.insert(0, "./torch-ext")