Skip to content

Commit 97a6411

Browse files
committed
chore: Moved macros to core/utils
Signed-off-by: Anurag Dixit <[email protected]>
1 parent be7eb30 commit 97a6411

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

core/conversion/conversionctx/ConversionCtx.h

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,10 +11,6 @@
1111
#include <cuda_runtime.h>
1212
#include "core/util/prelude.h"
1313

14-
#define DLA_SRAM_SIZE 1048576
15-
#define DLA_LOCAL_DRAM_SIZE 1073741824
16-
#define DLA_GLOBAL_DRAM_SIZE 536870912
17-
1814
namespace torch_tensorrt {
1915
namespace core {
2016
namespace conversion {

core/util/macros.h

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,11 @@
44

55
#define GET_MACRO(_1, _2, NAME, ...) NAME
66

7+
// DLA Memory related macros
8+
#define DLA_SRAM_SIZE 1048576
9+
#define DLA_LOCAL_DRAM_SIZE 1073741824
10+
#define DLA_GLOBAL_DRAM_SIZE 536870912
11+
712
#define TORCHTRT_LOG(l, sev, msg) \
813
do { \
914
std::stringstream ss{}; \

0 commit comments

Comments
 (0)