chore: Moved macros to core/utils

Signed-off-by: Anurag Dixit <a.dixit91@gmail.com>
pytorch · Aug 10, 2022 · ad7c081 · ad7c081
1 parent 5c646a3
commit ad7c081
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/core/conversion/conversionctx/ConversionCtx.h b/core/conversion/conversionctx/ConversionCtx.h
@@ -11,10 +11,6 @@
 #include <cuda_runtime.h>
 #include "core/util/prelude.h"
 
-#define DLA_SRAM_SIZE 1048576
-#define DLA_LOCAL_DRAM_SIZE 1073741824
-#define DLA_GLOBAL_DRAM_SIZE 536870912
-
 namespace torch_tensorrt {
 namespace core {
 namespace conversion {

diff --git a/core/util/macros.h b/core/util/macros.h
@@ -4,6 +4,11 @@
 
 #define GET_MACRO(_1, _2, NAME, ...) NAME
 
+// DLA Memory related macros
+#define DLA_SRAM_SIZE 1048576
+#define DLA_LOCAL_DRAM_SIZE 1073741824
+#define DLA_GLOBAL_DRAM_SIZE 536870912
+
 #define TORCHTRT_LOG(l, sev, msg) \
   do {                            \
     std::stringstream ss{};       \