pytorch
diff --git a/‎torch/_inductor/codegen/aoti_runtime/interface.cpp
Copy file name to clipboardExpand all lines: torch/_inductor/codegen/aoti_runtime/interface.cpp
+11Lines changed: 11 additions & 0 deletions b/‎torch/_inductor/codegen/aoti_runtime/interface.cpp
Copy file name to clipboardExpand all lines: torch/_inductor/codegen/aoti_runtime/interface.cpp
+11Lines changed: 11 additions & 0 deletions
diff --git a/‎torch/csrc/inductor/aoti_runtime/interface.h
Copy file name to clipboardExpand all lines: torch/csrc/inductor/aoti_runtime/interface.h
+8Lines changed: 8 additions & 0 deletions b/‎torch/csrc/inductor/aoti_runtime/interface.h
Copy file name to clipboardExpand all lines: torch/csrc/inductor/aoti_runtime/interface.h
+8Lines changed: 8 additions & 0 deletions
diff --git a/‎torch/csrc/inductor/aoti_runtime/model_container.h
Copy file name to clipboardExpand all lines: torch/csrc/inductor/aoti_runtime/model_container.h
+7Lines changed: 7 additions & 0 deletions b/‎torch/csrc/inductor/aoti_runtime/model_container.h
Copy file name to clipboardExpand all lines: torch/csrc/inductor/aoti_runtime/model_container.h
+7Lines changed: 7 additions & 0 deletions
@@ -207,6 +207,17 @@ AOTIRuntimeError AOTInductorModelContainerGetConstantDtype(
     { *dtype = container->constant_dtype(idx); })
 }
 
+AOTIRuntimeError AOTInductorModelContainerGetConstantDataSize(
+  AOTInductorModelContainerHandle container_handle,
+  size_t idx,
+  size_t* data_size) {
+  auto* container =
+    reinterpret_cast<torch::aot_inductor::AOTInductorModelContainer*>(
+        container_handle);
+  CONVERT_EXCEPTION_TO_ERROR_CODE(
+    { *data_size = container->constant_data_size(idx); })
+}
+
 AOTIRuntimeError AOTInductorModelContainerExtractConstantsMap(
     AOTInductorModelContainerHandle container_handle,
     AOTInductorConstantMapHandle constant_map_handle,
 
@@ -117,6 +117,14 @@ AOTIRuntimeError AOTInductorModelContainerGetConstantDtype(
     size_t idx,
     int32_t* dtype);
 
+// Retrieves a constant's data size.
+// idx is the index of the internal's constants.
+// Need idx < num_constants from AOTInductorModelContainerGetNumConstants
+AOTIRuntimeError AOTInductorModelContainerGetConstantDataSize(
+    AOTInductorModelContainerHandle container_handle,
+    size_t idx,
+    size_t* data_size);
+
 // Extract the constants that is being used in the container.
 AOTIRuntimeError AOTInductorModelContainerExtractConstantsMap(
     AOTInductorModelContainerHandle container_handle,
 
@@ -232,6 +232,13 @@ class AOTInductorModelContainer {
     return models_[0]->constant_from_folded(static_cast<int64_t>(idx));
   }
 
+  size_t constant_data_size(size_t idx) const {
+    if (this->num_models() == 0) {
+      throw std::runtime_error("No available models in container!");
+    }
+    return models_[0]->constant_data_size(static_cast<int64_t>(idx));
+  }
+
   // retrieve type of constants_info_[idx]
   int32_t constant_type(size_t idx) const {
     if (this->num_models() == 0) {