small fixes

Author: alonre24

.circleci/config.yml

@@ -239,7 +239,7 @@ jobs:
       - checkout-all
       - restore_cache:
           keys:
-          - v1-dependencies-{{ checksum "get_deps.sh" }}-<<parameters.osnick>>-<<parameters.target>>
+          - v1.2.5-deps-{{ checksum "get_deps.sh" }}-<<parameters.osnick>>-<<parameters.target>>
       - setup-automation
 
       # since we run in parallel, we need to generate docker files with different suffixes hence the DOCKER_SUFFIX

opt/build/docker/dockerfile-gpu-test.tmpl

@@ -11,8 +11,6 @@ ARG ARCH=x64
 # OS=ubuntu18.04|ubuntu16.04|centos7
 ARG OS=ubuntu18.04
 
-ARG CUDA_VER=11.2-cudnn8
-
 #----------------------------------------------------------------------------------------------
 FROM redisfab/redis:{{REDIS_VERSION}}-{{REDIS_ARCH}}-{{REDIS_OSNICK}} AS redis
 FROM nvidia/cuda:{{REDIS_CUDA_VERSION}}-devel-ubuntu18.04 AS builder

opt/build/onnxruntime/Dockerfile.x64-gpu

@@ -7,9 +7,8 @@ export REDIS_BAZEL_VERSION
 PRODUCT=tensorflow
 DOCKER_ORG=redislabs
 VERSION=${REDIS_TF_VERSION}
-REDIS_CUDA_VERSION=11.0-cudnn8
+REDIS_CUDA_VERSION=11.2.2-cudnn8
 
-export REDIS_ONNX_VERSION
 OSNICK=xenial
 
 ROOT=.

opt/build/tensorflow/Makefile

@@ -35,4 +35,4 @@ std::vector<torch::Tensor> modelExecute(const std::string &model_key,
                                         int64_t num_outputs);
 
 // Register Redis and RedisAI costume ops in torch
-void registerRedisOps(void);
+void registerRedisOps(void);