Minor style fixes for sgl-kernel (#9289)
This commit is contained in:
@@ -33,6 +33,7 @@ TORCH_LIBRARY_EXPAND(sgl_kernel, m) {
|
||||
|
||||
m.def("gelu_quick(Tensor! out, Tensor input) -> ()");
|
||||
m.impl("gelu_quick", torch::kCUDA, &gelu_quick);
|
||||
|
||||
/*
|
||||
* From csrc/allreduce
|
||||
*/
|
||||
|
||||
Reference in New Issue
Block a user