Spaces:
Runtime error
Runtime error
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 (
|
21 |
}
|
|
|
17 |
}
|
18 |
|
19 |
PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
|
20 |
+
m.def("fused_bias_act", &fused_bias_act, "fused bias act (cuda)");
|
21 |
}
|