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 |
}
|