onnx info

Author: DvirDukhan

src/backends.c

@@ -417,6 +417,16 @@ int RAI_LoadBackend_ONNXRuntime(RedisModuleCtx *ctx, const char *path) {
         return REDISMODULE_ERR;
     }
 
+    backend.get_version = (const char*(*)(void))(unsigned long)dlsym(handle, "RAI_GetBackendVersionORT");
+    if (backend.get_version == NULL) {
+        dlclose(handle);
+        RedisModule_Log(ctx, "warning",
+                        "Backend does not export RAI_GetBackendVersionORT. ONNX backend "
+                        "not loaded from %s",
+                        path);
+        return REDISMODULE_ERR;
+    }
+
     RAI_backends.onnx = backend;
 
     RedisModule_Log(ctx, "notice", "ONNX backend loaded from %s", path);

src/backends/onnxruntime.c

@@ -670,3 +670,8 @@ int RAI_ModelSerializeORT(RAI_Model *model, char **buffer, size_t *len, RAI_Erro
 
     return 0;
 }
+
+const char* RAI_GetBackendVersionORT(void) {
+    return OrtGetApiBase()->GetVersionString();
+}
+

src/backends/onnxruntime.h

@@ -17,4 +17,6 @@ int RAI_ModelRunORT(RAI_ModelRunCtx **mctxs, RAI_Error *error);
 
 int RAI_ModelSerializeORT(RAI_Model *model, char **buffer, size_t *len, RAI_Error *error);
 
+const char* RAI_GetBackendVersionORT(void);
+
 #endif /* SRC_BACKENDS_ONNXRUNTIME_H_ */

src/redisai.c

@@ -847,9 +847,9 @@ void _RedisAI_Info(RedisModuleCtx* ctx) {
         RedisModule_ReplyWithSimpleString(ctx, RAI_backends.tf.get_version()); 
     }
 
-    if(RAI_backends.torch.get_version){
-        RedisModule_ReplyWithSimpleString(ctx, "Torch version"); 
-        RedisModule_ReplyWithSimpleString(ctx, RAI_backends.torch.get_version()); 
+    if(RAI_backends.onnx.get_version){
+        RedisModule_ReplyWithSimpleString(ctx, "ONNX version"); 
+        RedisModule_ReplyWithSimpleString(ctx, RAI_backends.onnx.get_version()); 
     }