easyconfigs-it4i/t/Tensorflow/TensorFlow-2.9.1_remove-duplicate-gpu-tests.patch
easybuild 3178cb7962 new file: f/flatbuffers/flatbuffers-2.0.7-GCCcore-11.3.0.eb
new file:   o/OpenMPI/OpenMPI-4.1.4-GCC-12.2.0.eb
	modified:   q/QuantumESPRESSO/QuantumESPRESSO-7.1-NVHPC-21.9.eb
	new file:   t/Tensorflow/TensorFlow-2.10.1-foss-2022a-CUDA-11.7.0.eb
	new file:   t/Tensorflow/TensorFlow-2.8.4_exclude-xnnpack-on-ppc.patch
	new file:   t/Tensorflow/TensorFlow-2.8.4_resolve-gcc-symlinks.patch
	new file:   t/Tensorflow/TensorFlow-2.9.1_remove-duplicate-gpu-tests.patch
	new file:   t/Tensorflow/TensorFlow-2.9.1_remove-libclang-and-io-gcs-deps.patch
	new file:   u/UCX/UCX-1.13.1-GCCcore-12.2.0.eb
	new file:   x/XALT/XALT-2.10.45-GPU-eb.eb
	modified:   x/XALT/XALT-2.10.45-eb.eb
2023-01-25 17:48:29 +01:00

56 lines
1.7 KiB
Diff

TensorFlow adds some GPU tests twice increasing the runtime of the test suite.
This filters out the test part meant for CPU.
See https://github.com/tensorflow/tensorflow/issues/47081
From https://github.com/tensorflow/tensorflow/pull/59129
Author: Alexander Grund (TU Dresden)
---
tensorflow/tensorflow.bzl | 33 +++++++++++++++++----------------
1 file changed, 17 insertions(+), 16 deletions(-)
diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl
index 649c8e22dcc95..d3fc0e3221497 100644
--- a/tensorflow/tensorflow.bzl
+++ b/tensorflow/tensorflow.bzl
@@ -1461,22 +1461,23 @@ def tf_gpu_cc_test(
linkopts = [],
**kwargs):
targets = []
- tf_cc_test(
- name = name,
- size = size,
- srcs = srcs,
- args = args,
- data = data,
- extra_copts = extra_copts + if_cuda(["-DNV_CUDNN_DISABLE_EXCEPTION"]),
- kernels = kernels,
- linkopts = linkopts,
- linkstatic = linkstatic,
- suffix = "_cpu",
- tags = tags,
- deps = deps,
- **kwargs
- )
- targets.append(name + "_cpu")
+ if 'gpu' not in tags:
+ tf_cc_test(
+ name = name,
+ size = size,
+ srcs = srcs,
+ args = args,
+ data = data,
+ extra_copts = extra_copts + if_cuda(["-DNV_CUDNN_DISABLE_EXCEPTION"]),
+ kernels = kernels,
+ linkopts = linkopts,
+ linkstatic = linkstatic,
+ suffix = "_cpu",
+ tags = tags,
+ deps = deps,
+ **kwargs
+ )
+ targets.append(name + "_cpu")
tf_cc_test(
name = name,
size = size,