Skip to content

Commit bb7cbbb

Browse files
ndonyapourmr-c
authored andcommitted
add cudaDeviceCount flag to the request requirements
1 parent 509ffb9 commit bb7cbbb

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

cwltool/process.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1002,6 +1002,9 @@ def evalResources(
10021002
request[a + "Min"] = mn
10031003
request[a + "Max"] = cast(Union[int, float], mx)
10041004

1005+
if cudaReq:
1006+
request["cudaDeviceCount"] = request["cudaDeviceCountMin"]
1007+
10051008
request_evaluated = cast(Dict[str, Union[int, float]], request)
10061009
if runtimeContext.select_resources is not None:
10071010
# Call select resources hook

0 commit comments

Comments
 (0)