refdctor: Changed the hardcoded values to macros

Signed-off-by: Anurag Dixit <[email protected]>

Author: andi4191

core/conversion/conversionctx/ConversionCtx.cpp

@@ -124,13 +124,13 @@ ConversionCtx::ConversionCtx(BuilderSettings build_settings)
         settings.enabled_precisions.find(nvinfer1::DataType::kFLOAT) == settings.enabled_precisions.end(),
         "DLA supports only fp16 or int8 precision");
     cfg->setDLACore(settings.device.dla_core);
-    if (settings.dla_sram_size != 1048576) {
+    if (settings.dla_sram_size != DLA_SRAM_SIZE) {
       cfg->setMemoryPoolLimit(nvinfer1::MemoryPoolType::kDLA_MANAGED_SRAM, settings.dla_sram_size);
     }
-    if (settings.dla_local_dram_size != 1073741824) {
+    if (settings.dla_local_dram_size != DLA_LOCAL_DRAM_SIZE) {
       cfg->setMemoryPoolLimit(nvinfer1::MemoryPoolType::kDLA_LOCAL_DRAM, settings.dla_local_dram_size);
     }
-    if (settings.dla_global_dram_size != 536870912) {
+    if (settings.dla_global_dram_size != DLA_GLOBAL_DRAM_SIZE) {
       cfg->setMemoryPoolLimit(nvinfer1::MemoryPoolType::kDLA_GLOBAL_DRAM, settings.dla_global_dram_size);
     }
   }

core/conversion/conversionctx/ConversionCtx.h

@@ -11,6 +11,10 @@
 #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 {
@@ -35,9 +39,9 @@ struct BuilderSettings {
   nvinfer1::IInt8Calibrator* calibrator = nullptr;
   uint64_t num_avg_timing_iters = 1;
   uint64_t workspace_size = 0;
-  uint64_t dla_sram_size = 1048576;
-  uint64_t dla_local_dram_size = 1073741824;
-  uint64_t dla_global_dram_size = 536870912;
+  uint64_t dla_sram_size = DLA_SRAM_SIZE;
+  uint64_t dla_local_dram_size = DLA_LOCAL_DRAM_SIZE;
+  uint64_t dla_global_dram_size = DLA_GLOBAL_DRAM_SIZE;
 
   BuilderSettings() = default;
   BuilderSettings(const BuilderSettings& other) = default;