Skip to content

Commit 3d6ed3a

Browse files
fix: fix name of gpu_manufacturer resource (#35)
<!-- This is an auto-generated comment: release notes by coderabbit.ai --> ## Summary by CodeRabbit - **Bug Fixes** - Improved GPU resource detection to ensure accurate allocation based on manufacturer, maintaining consistent support for both NVIDIA and AMD devices. <!-- end of auto-generated comment: release notes by coderabbit.ai -->
1 parent 4d54b2c commit 3d6ed3a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

snakemake_executor_plugin_kubernetes/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -329,7 +329,7 @@ def run_job(self, job: JobExecutorInterface):
329329
# But let's keep nvidia.com/gpu
330330
# for both if the cluster doesn't differentiate.
331331
# If your AMD plugin uses a different name, update accordingly:
332-
manufacturer = resources_dict.get("manufacturer", "").lower()
332+
manufacturer = resources_dict.get("gpu_manufacturer", "").lower()
333333
if manufacturer == "nvidia":
334334
container.resources.requests["nvidia.com/gpu"] = gpu_count
335335
self.logger.debug(f"Requested NVIDIA GPU resources: {gpu_count}")

0 commit comments

Comments
 (0)