Fix a bug that passing bytes size instead of kb.

0ctopus13prime · Sep 16, 2024 · 4238b0b · 4238b0b
1 parent 6cdde7f
commit 4238b0b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/main/java/org/opensearch/knn/index/KNNIndexShard.java b/src/main/java/org/opensearch/knn/index/KNNIndexShard.java
@@ -97,7 +97,8 @@ public void warmup() throws IOException {
                     nativeMemoryCacheManager.get(
                         new NativeMemoryEntryContext.IndexEntryContext(
                             directory,
-                            engineFileContext.getIndexPath(),
+                            // engineFileContext.getIndexPath(), TMP
+                            engineFileContext.getIndexPath().substring(engineFileContext.getIndexPath().lastIndexOf("/") + 1),
                             NativeMemoryLoadStrategy.IndexLoadStrategy.getInstance(),
                             getParametersAtLoading(
                                 engineFileContext.getSpaceType(),

diff --git a/src/main/java/org/opensearch/knn/index/memory/NativeMemoryLoadStrategy.java b/src/main/java/org/opensearch/knn/index/memory/NativeMemoryLoadStrategy.java
@@ -153,7 +153,7 @@ public NativeMemoryAllocation.IndexAllocation load(NativeMemoryEntryContext.Inde
                 return new NativeMemoryAllocation.IndexAllocation(
                     executor,
                     indexAddress,
-                    indexSize,
+                    indexSize / 1024,
                     knnEngine,
                     indexPath,
                     indexEntryContext.getOpenSearchIndexName(),