Fix test for GPU+CPU

Author: alonre24

opt/build/onnxruntime/Dockerfile.x64

@@ -24,7 +24,7 @@ RUN apt-get install -y locales && \
 
 WORKDIR /build
 
-ARG BUILDTYPE=Debug
+ARG BUILDTYPE=MinSizeRel
 
 ARG BUILDARGS="--config ${BUILDTYPE} --parallel"
 

opt/build/onnxruntime/Dockerfile.x64-gpu

@@ -40,4 +40,4 @@ RUN ./build.sh ${BUILDARGS} --update --build --use_cuda --cudnn_home /usr/local/
 
 ADD ./pack.sh /build/
 WORKDIR /build
-RUN /pack.sh ${ONNXRUNTIME_VER} ${ARCH}
+RUN ./pack.sh ${ONNXRUNTIME_VER} ${ARCH}

opt/build/onnxruntime/pack.sh

@@ -5,7 +5,7 @@ VER="$1"
 PLATFORM="$2"
 
 mkdir -p pack/include pack/lib
-cp onnxruntime/build/Linux/Debug/libonnxruntime.so.${VER} pack/lib/
+cp onnxruntime/build/Linux/MinSizeRel/libonnxruntime.so.${VER} pack/lib/
 cp onnxruntime/docs/C_API_Guidelines.md pack/
 cp onnxruntime/LICENSE pack/
 cp onnxruntime/README.md pack/

src/backends/onnxruntime.c

@@ -32,6 +32,7 @@ const OrtMemoryInfo *AllocatorInfo(const OrtAllocator *allocator) {
     return mem_info;
 }
 
+// Allocate address with 64-byte alignment to cope with onnx optimizations.
 void *AllocatorAlloc(OrtAllocator *ptr, size_t size) {
 
     (void)ptr;
@@ -47,8 +48,9 @@ void *AllocatorAlloc(OrtAllocator *ptr, size_t size) {
 
 void AllocatorFree(OrtAllocator *ptr, void *p) {
     (void)ptr;
-    if (p == NULL)
+    if (p == NULL) {
         return;
+    }
     void *p1 = ((void **)p)[-1];
     size_t allocated_size = RedisModule_MallocSize(p1);
     atomic_fetch_sub(&OnnxMemory, allocated_size);

tests/flow/tests_onnx.py

@@ -538,5 +538,5 @@ def test_onnx_use_custom_allocator_with_GPU(env):
     ai_memory_config = {k.split(":")[0]: k.split(":")[1]
                         for k in con.execute_command("INFO MODULES").decode().split("#")[4].split()[1:]}
     env.assertTrue(int(ai_memory_config["ai_onnxruntime_memory"]) > 100)
-    env.assertTrue(int(ai_memory_config["ai_onnxruntime_memory"]) < 400)
+    env.assertTrue(int(ai_memory_config["ai_onnxruntime_memory"]) < 705)
     env.assertEqual(int(ai_memory_config["ai_onnxruntime_memory_access_num"]), 5)