Merge pull request #168 from Duke-GCB/issue-167

adds 'cudaVersionMin', 'cudaComputeCapability' in CUDARequirement
Duke-GCB · May 2, 2024 · 753a523 · 753a523
2 parents f633264 + ae81c0f
commit 753a523
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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):