Move scheduling op bindings to a separate file (#3639)

NVIDIA · Jan 2, 2025 · 5ed5dad · 5ed5dad
1 parent ed5b3c9
commit 5ed5dad
Show file tree

Hide file tree

Showing 5 changed files with 533 additions and 505 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -443,6 +443,7 @@ if(BUILD_PYTHON)
   list(APPEND NVFUSER_PYTHON_SRCS
     ${NVFUSER_SRCS_DIR}/python_frontend/python_bindings.cpp
     ${NVFUSER_SRCS_DIR}/python_frontend/python_bindings_extension.cpp
+    ${NVFUSER_SRCS_DIR}/python_frontend/schedule_bindings.cpp
   )
 
   add_library(nvf_py_internal OBJECT ${NVFUSER_PYTHON_SRCS})

diff --git a/csrc/python_frontend/fusion_definition.h b/csrc/python_frontend/fusion_definition.h
@@ -6,14 +6,15 @@
  */
 // clang-format on
 #pragma once
+
 #include <exceptions.h>
 #include <functional>
 #include <iostream>
+#include <unordered_map>
 
 #include <python_frontend/fusion_state.h>
 #include <python_frontend/segmentation.h>
 #include <visibility.h>
-#include <unordered_map>
 
 namespace nvfuser::python_frontend {