feng2022 commited on
Commit
fccf9f3
1 Parent(s): bd964f4

Update Time_TravelRephotography/op/fused_act.py

Browse files
Time_TravelRephotography/op/fused_act.py CHANGED
@@ -7,13 +7,13 @@ from torch.utils.cpp_extension import load
7
 
8
 
9
  module_path = os.path.dirname(__file__)
10
- fused = load(
11
- 'fused',
12
- sources=[
13
- os.path.join(module_path, 'fused_bias_act.cpp'),
14
- os.path.join(module_path, 'fused_bias_act_kernel.cu'),
15
- ],
16
- )
17
 
18
 
19
  class FusedLeakyReLUFunctionBackward(Function):
 
7
 
8
 
9
  module_path = os.path.dirname(__file__)
10
+ # fused = load(
11
+ # 'fused',
12
+ # sources=[
13
+ # os.path.join(module_path, 'fused_bias_act.cpp'),
14
+ # os.path.join(module_path, 'fused_bias_act_kernel.cu'),
15
+ # ],
16
+ # )
17
 
18
 
19
  class FusedLeakyReLUFunctionBackward(Function):