use PYBIND11_MODULE in global namespace

chengzeyi · Nov 22, 2023 · 89697a0 · 89697a0
1 parent bd79514
commit 89697a0
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/sfast/csrc/main.cpp b/sfast/csrc/main.cpp
@@ -7,7 +7,7 @@
 #include "operators/cutlass/cutlass_qlinear.h"
 #include "operators/fused_linear.h"
 
-namespace sfast {
+using namespace sfast;
 
 PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
   jit::initJITBindings(m);
@@ -19,6 +19,4 @@ TORCH_LIBRARY(sfast, m) {
   operators::initCUBLASGEMMBindings(m);
   operators::initCutlassQLinearBindings(m);
   operators::initFusedLinearBindings(m);
-}
-
-} // namespace sfast
+}