feat(chart): install gpu-sharing as OCI dependency

Signed-off-by: davidLif <davidshani12@gmail.com>

Author: davidLif

.changes/unreleased/added-20260810-112731.yaml

@@ -0,0 +1,3 @@
+kind: Added
+body: |-
+  Install gpu-sharing as an OCI subchart

.github/workflows/on-pr.yaml

@@ -233,6 +233,7 @@ jobs:
 
       - name: Build helm chart
         run: |
+          helm dependency build ./deployments/kai-scheduler
           helm package ./deployments/kai-scheduler -d ./charts --app-version $PACKAGE_VERSION --version $PACKAGE_VERSION
           cp charts/kai-scheduler-$PACKAGE_VERSION.tgz /mnt/images/
 

.github/workflows/on-release.yaml

@@ -63,6 +63,7 @@ jobs:
       - name: Build helm chart
         run: |
           sed -i 's#registry/local/kai-scheduler#${{ env.DOCKER_REGISTRY }}#' deployments/kai-scheduler/values.yaml
+          helm dependency build ./deployments/kai-scheduler
           helm package ./deployments/kai-scheduler -d ./charts --app-version $PACKAGE_VERSION --version $PACKAGE_VERSION
           cp charts/kai-scheduler-$PACKAGE_VERSION.tgz /mnt/images/
 

.github/workflows/push-artifacts.yaml

@@ -59,6 +59,7 @@ jobs:
       - name: Build helm chart
         run: |
           sed -i 's#registry/local/kai-scheduler#${{ env.DOCKER_REGISTRY }}#' deployments/kai-scheduler/values.yaml
+          helm dependency build ./deployments/kai-scheduler
           helm package ./deployments/kai-scheduler -d ./charts --app-version $PACKAGE_VERSION --version $PACKAGE_VERSION
 
       - name: Push Helm Chart

Makefile

@@ -27,8 +27,13 @@ K8S_COPYRIGHTED_MANIFEST_FILES := deployments/kai-scheduler/crds/kai.scheduler_t
 lint: fmt-go vet-go lint-go
 .PHONY: lint
 
+.PHONY: chart-deps
+chart-deps:
+	@echo "Fetching Helm chart dependencies: kai-scheduler"
+	helm dependency build ./deployments/kai-scheduler
+
 .PHONY: test-chart
-test-chart:
+test-chart: chart-deps
 	@echo "Running tests for Helm chart: kai-scheduler"
 	docker run -t --rm -v ./deployments/kai-scheduler:/apps helmunittest/helm-unittest:3.17.2-0.8.1 . -f 'tests/**/*_test.yaml'
 

deployments/kai-scheduler/Chart.lock

@@ -0,0 +1,6 @@
+dependencies:
+- name: gpu-sharing
+  repository: oci://ghcr.io/kai-scheduler/gpu-sharing
+  version: v0.1.0-alpha.1
+digest: sha256:c3d7f76c6d2b2fd7b256b7450fb435a82693e8acef1265d10a8d5270fcb028db
+generated: "2026-08-02T20:53:54.374149+03:00"

deployments/kai-scheduler/Chart.yaml

@@ -6,4 +6,10 @@ name: kai-scheduler
 description: KAI Scheduler by NVIDIA
 type: application
 version: 0.0.0
-appVersion: 0.0.0
+appVersion: 0.0.0
+
+dependencies:
+  - name: gpu-sharing
+    version: v0.1.0-alpha.1
+    repository: oci://ghcr.io/kai-scheduler/gpu-sharing
+    condition: global.nvFractions.set

deployments/kai-scheduler/templates/_helpers.tpl

@@ -108,10 +108,14 @@ spec:
     {{- if .Values.podgrouper.queueLabelKey }}
     queueLabelKey: {{ .Values.podgrouper.queueLabelKey | quote }}
     {{- end }}
-    {{- /* GpuSharingMode precedence: an explicit (non-empty) mode wins; otherwise a
-           legacy gpuSharing value is passed through (operator derives the mode); when
-           neither is set, default to NonMemoryEnforced. */}}
-    {{- if .Values.global.gpuSharingMode }}
+    {{- /* GpuSharingMode precedence: nvFractions.set forces NvFractions (validate.yaml has
+           already rejected a conflicting explicit mode, so this only guards against a legacy
+           gpuSharing value downgrading it); then an explicit (non-empty) mode; then a legacy
+           gpuSharing value is passed through (operator derives the mode); when none is set,
+           default to NonMemoryEnforced. */}}
+    {{- if .Values.global.nvFractions.set }}
+    gpuSharingMode: NvFractions
+    {{- else if .Values.global.gpuSharingMode }}
     gpuSharingMode: {{ .Values.global.gpuSharingMode }}
     {{- else if not (hasKey .Values.global "gpuSharing") }}
     gpuSharingMode: NonMemoryEnforced
@@ -260,7 +264,7 @@ spec:
         maxUnavailable: {{ .Values.admission.podDisruptionBudget.maxUnavailable }}
         {{- end }}
       {{- end }}
-    {{- if and (not .Values.global.gpuSharingMode) (hasKey .Values.global "gpuSharing") }}
+    {{- if and (not .Values.global.nvFractions.set) (not .Values.global.gpuSharingMode) (hasKey .Values.global "gpuSharing") }}
     gpuSharing: {{ .Values.global.gpuSharing }}
     {{- end }}
     blockNvidiaVisibleDevices: {{ .Values.global.blockNvidiaVisibleDevices | default false }}

deployments/kai-scheduler/templates/validate.yaml

@@ -0,0 +1,9 @@
+{{/*
+Copyright 2025 NVIDIA CORPORATION
+SPDX-License-Identifier: Apache-2.0
+*/}}
+{{- /* Renders no resource. Aborts the render when nvFractions.set asks for the gpu-sharing
+       operator while gpuSharingMode explicitly selects a different mode. */}}
+{{- if and .Values.global.nvFractions.set .Values.global.gpuSharingMode (ne .Values.global.gpuSharingMode "NvFractions") }}
+{{- fail (printf "global.nvFractions.set=true requires global.gpuSharingMode to be empty or \"NvFractions\", got %q" .Values.global.gpuSharingMode) }}
+{{- end }}

deployments/kai-scheduler/tests/gpu_sharing_mode_test.yaml

@@ -4,8 +4,8 @@
 suite: test gpu sharing mode configuration
 
 set:
-  _test:
-    renderKaiConfig: true
+  kaiConfigDeployer.enabled: false
+  kaiConfig.render: true
 templates:
   - kai-config.yaml
 tests: