Move BaseGenerator logic to core (#7)

cosdt · Jul 16, 2024 · c1b2f0a · c1b2f0a
1 parent 934e6e5
commit c1b2f0a
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 6 deletions.
diff --git a/csrc/npu/BaseGeneratorImpl.cpp → csrc/core/BaseGeneratorImpl.cpp b/csrc/npu/BaseGeneratorImpl.cpp → csrc/core/BaseGeneratorImpl.cpp
diff --git a/csrc/npu/BaseGeneratorImpl.h → csrc/core/BaseGeneratorImpl.h b/csrc/npu/BaseGeneratorImpl.h → csrc/core/BaseGeneratorImpl.h
diff --git a/csrc/npu/NPUGeneratorImpl.cpp b/csrc/npu/NPUGeneratorImpl.cpp
@@ -1,12 +1,9 @@
+#include "NPUGeneratorImpl.h"
 #include <ATen/Utils.h>
 #include <ATen/core/GeneratorForPrivateuseone.h>
 #include <c10/core/StreamGuard.h>
-
-#include "csrc/npu/NPUFunctions.h"
-
-#include "BaseGeneratorImpl.h"
-#include "NPUGeneratorImpl.h"
 #include "csrc/aten/generated/NPUNativeFunctions.h"
+#include "csrc/npu/NPUFunctions.h"
 
 namespace at_npu {
 namespace detail {

diff --git a/csrc/npu/NPUGeneratorImpl.h b/csrc/npu/NPUGeneratorImpl.h
@@ -5,7 +5,7 @@
 #include <ATen/core/Generator.h>
 #include <c10/core/GeneratorImpl.h>
 #include <limits>
-#include "BaseGeneratorImpl.h"
+#include "csrc/core/BaseGeneratorImpl.h"
 #include "npu/core/NPUMacros.h"
 
 namespace at_npu {