From 4fb7ce56b49881477717650b4a09f3568f223dd3 Mon Sep 17 00:00:00 2001
From: Fabrice Brito <fabrice.brito@terradue.com>
Date: Thu, 2 May 2024 23:10:05 +0200
Subject: [PATCH] adds 'cudaVersionMin', 'cudaComputeCapability' in
 CUDARequirement

---
 calrissian/job.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/calrissian/job.py b/calrissian/job.py
index ce08e24..ee02f68 100644
--- a/calrissian/job.py
+++ b/calrissian/job.py
@@ -481,7 +481,9 @@ def finish(self, completion_result: CompletionResult, runtimeContext):
     # Not yet complete, only checks features of DockerRequirement
     supported_features = {
         'DockerRequirement': ['class', 'dockerPull'],
-        'http://commonwl.org/cwltool#CUDARequirement': ['class', 'cudaDeviceCount', 'cudaDeviceCountMin', 'cudaDeviceCountMax'],
+        'http://commonwl.org/cwltool#CUDARequirement': ['class', 'cudaDeviceCount', 
+                                                        'cudaDeviceCountMin', 'cudaDeviceCountMax', 
+                                                        'cudaVersionMin', 'cudaComputeCapability'],
     }
 
     def check_requirements(self, runtimeContext):