feng2022 commited on
Commit
fadf666
1 Parent(s): e53e937

Update Time_TravelRephotography/op/fused_bias_act.cpp

Browse files
Time_TravelRephotography/op/fused_bias_act.cpp CHANGED
@@ -17,5 +17,5 @@ torch::Tensor fused_bias_act(const torch::Tensor& input, const torch::Tensor& bi
17
  }
18
 
19
  PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
20
- m.def("fused_bias_act", &fused_bias_act, "fused bias act (cpu)");
21
  }
 
17
  }
18
 
19
  PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
20
+ m.def("fused_bias_act", &fused_bias_act, "fused bias act (CUDA)");
21
  }