log the number milliseconds that other backends are running as well

Author: alonre24

src/backends/tensorflow.c

@@ -1,4 +1,6 @@
 #define REDISMODULE_MAIN
+
+#include <redis_ai_objects/stats.h>
 #include "backends/util.h"
 #include "backends/tensorflow.h"
 #include "util/arr.h"
@@ -14,6 +16,7 @@ int RAI_InitBackendTF(int (*get_api_fn)(const char *, void *)) {
     get_api_fn("RedisModule_Free", ((void **)&RedisModule_Free));
     get_api_fn("RedisModule_Realloc", ((void **)&RedisModule_Realloc));
     get_api_fn("RedisModule_Strdup", ((void **)&RedisModule_Strdup));
+    get_api_fn("RedisModule_Log", ((void **)&RedisModule_Log));
 
     return REDISMODULE_OK;
 }
@@ -520,10 +523,12 @@ int RAI_ModelRunTF(RAI_Model *model, RAI_ExecutionCtx **ectxs, RAI_Error *error)
         }
         outputs[i] = port;
     }
-
+    long long time_before = mstime();
     TF_SessionRun(tfSession, NULL /* run_options */, inputs, inputTensorsValues, ninputs, outputs,
                   outputTensorsValues, noutputs, NULL /* target_opers */, 0 /* ntargets */,
                   NULL /* run_Metadata */, status);
+    long long time_after = mstime();
+    RedisModule_Log(NULL, "notice", "tf run time was: %lld", time_after - time_before);
 
     for (size_t i = 0; i < ninputs; ++i) {
         TF_DeleteTensor(inputTensorsValues[i]);

src/backends/torch.c

@@ -1,4 +1,6 @@
 #define REDISMODULE_MAIN
+
+#include <redis_ai_objects/stats.h>
 #include "backends/util.h"
 #include "backends/torch.h"
 #include "util/arr.h"
@@ -29,6 +31,7 @@ int RAI_InitBackendTorch(int (*get_api_fn)(const char *, void *)) {
                ((void **)&RedisModule_ThreadSafeContextUnlock));
     get_api_fn("RedisModule_FreeThreadSafeContext", ((void **)&RedisModule_FreeThreadSafeContext));
     get_api_fn("RedisModule_StringPtrLen", ((void **)&RedisModule_StringPtrLen));
+    get_api_fn("RedisModule_Log", ((void **)&RedisModule_Log));
 
     return REDISMODULE_OK;
 }
@@ -228,7 +231,10 @@ int RAI_ModelRunTorch(RAI_Model *model, RAI_ExecutionCtx **ectxs, RAI_Error *err
     }
 
     char *error_descr = NULL;
+    long long time_before = mstime();
     torchRunModel(RAI_ModelGetModel(model), ninputs, inputs_dl, noutputs, outputs_dl, &error_descr);
+    long long time_after = mstime();
+    RedisModule_Log(NULL, "notice", "torch run time was: %lld", time_after - time_before);
 
     for (size_t i = 0; i < ninputs; ++i) {
         RAI_TensorFree(inputs[i]);