Configure Time slicing for NVidia GPU e2e PR check

.github/workflows/e2e_tests.yaml

@@ -38,8 +38,8 @@ jobs:
       - name: Checkout common repo code
         uses: actions/checkout@v4
         with:
-          repository: 'project-codeflare/codeflare-common'
-          ref: 'main'
+          repository: 'sutaakar/codeflare-common'
+          ref: 'gpu-time-slicing'
           path: 'common'
 
       - name: Set Go
@@ -61,6 +61,8 @@ jobs:
 
       - name: Install NVidia GPU operator for KinD
         uses: ./common/github-actions/nvidia-gpu-operator
+        with:
+          enable-time-slicing: true
 
       - name: Deploy CodeFlare stack
         id: deploy
@@ -115,7 +117,7 @@ jobs:
 
       - name: Upload logs
         uses: actions/upload-artifact@v4
-        if: always() && steps.deploy.outcome == 'success'
+        if: always() && steps.kind-install.outcome == 'success'
         with:
           name: logs
           retention-days: 10

test/e2e/mnist_pytorch_appwrapper_test.go

@@ -41,6 +41,7 @@ func TestMnistPyTorchAppWrapperGpu(t *testing.T) {
 // Trains the MNIST dataset as a batch Job in an AppWrapper, and asserts successful completion of the training job.
 func runMnistPyTorchAppWrapper(t *testing.T, accelerator string) {
 	test := With(t)
+	test.T().Parallel()
 
 	// Create a namespace and localqueue in that namespace
 	namespace := test.NewTestNamespace()

test/e2e/mnist_rayjob_raycluster_test.go

@@ -48,6 +48,7 @@ func TestMnistRayJobRayClusterGpu(t *testing.T) {
 
 func runMnistRayJobRayCluster(t *testing.T, accelerator string, numberOfGpus int) {
 	test := With(t)
+	test.T().Parallel()
 
 	// Create a namespace and localqueue in that namespace
 	namespace := test.NewTestNamespace()
@@ -108,6 +109,7 @@ func TestMnistRayJobRayClusterAppWrapperGpu(t *testing.T) {
 // Same as TestMNISTRayJobRayCluster, except the RayCluster is wrapped in an AppWrapper
 func runMnistRayJobRayClusterAppWrapper(t *testing.T, accelerator string, numberOfGpus int) {
 	test := With(t)
+	test.T().Parallel()
 
 	// Create a namespace and localqueue in that namespace
 	namespace := test.NewTestNamespace()

test/e2e/setup.sh

@@ -93,5 +93,5 @@ spec:
       - name: "memory"
         nominalQuota: "20G"
       - name: "nvidia.com/gpu"
-        nominalQuota: "1"
+        nominalQuota: "4"
 EOF