From 14e19e1e13ab44f9fcf391aa99e878c87c24b644 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20Laurent?= <francois.laurent@posteo.net>
Date: Mon, 17 Apr 2023 22:23:37 +0200
Subject: [PATCH] setting CUBLAS_WORKSPACE_CONFIG for reproducibility is the
default
---
scripts/larvatagger-toolkit.jl | 2 ++
test/scenarii.sh | 10 +++++-----
2 files changed, 7 insertions(+), 5 deletions(-)
diff --git a/scripts/larvatagger-toolkit.jl b/scripts/larvatagger-toolkit.jl
index 0898899..6ba92b7 100755
--- a/scripts/larvatagger-toolkit.jl
+++ b/scripts/larvatagger-toolkit.jl
@@ -9,6 +9,8 @@ fi
FLAGS=
if [ "$1" = "--sysimage" -o "$1" = "-J" ]; then FLAGS="--sysimage $2 "; shift 2; fi
if [ "${1:0:2}" = "-J" ]; then FLAGS="$1"; shift; fi
+if [ "$1" = train ] && [ -n "$(which nvidia-smi)" ] && [ -z "$CUBLAS_WORKSPACE_CONFIG" ];
+then export CUBLAS_WORKSPACE_CONFIG=:4096:8; fi
if [ -z "$JULIA" ]; then JULIA=julia; fi
exec $JULIA --project="$PROJECT_DIR" --color=yes --startup-file=no $FLAGS\
"${BASH_SOURCE[0]}" "$@"
diff --git a/test/scenarii.sh b/test/scenarii.sh
index 2f577b4..5f8932a 100755
--- a/test/scenarii.sh
+++ b/test/scenarii.sh
@@ -18,11 +18,11 @@ if [ -z "$JULIA_PROJECT" ]; then
fi
if [ -n "$(which nvidia-smi)" ]; then
- if [ -z "$CUBLAS_WORKSPACE_CONFIG" ]; then
- echo "Setting CUBLAS_WORKSPACE_CONFIG=:4096:8 to ensure reproducibility;"
- echo "see https://docs.nvidia.com/cuda/cublas/index.html#results-reproducibility"
- export CUBLAS_WORKSPACE_CONFIG=:4096:8
- fi
+ # if [ -z "$CUBLAS_WORKSPACE_CONFIG" ]; then
+ # echo "Setting CUBLAS_WORKSPACE_CONFIG=:4096:8 to ensure reproducibility;"
+ # echo "see https://docs.nvidia.com/cuda/cublas/index.html#results-reproducibility"
+ # export CUBLAS_WORKSPACE_CONFIG=:4096:8
+ # fi
tagger_suffix=_gpu
fi
--
GitLab