xpk 1.1.1__tar.gz → 1.1.2__tar.gz

{xpk-1.1.1/src/xpk.egg-info → xpk-1.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xpk
-Version: 1.1.1
+Version: 1.1.2
 Summary: xpk helps Cloud developers to orchestrate training jobs on accelerators on GKE.
 Author-email: XPK team <xpk-code-reviewers@google.com>
 License: Apache-2.0

{xpk-1.1.1 → xpk-1.1.2}/recipes/Workload_create_super-slicing.md

@@ -47,7 +47,7 @@ docker buildx build --platform=linux/amd64 -f 4b6736a12db8ea0f78ce793fd0d4ee0c94
 docker tag dry-run-runner gcr.io/golden-project/dry-run-runner:prefix-current
 [XPK] Task: `Upload Docker Image` is implemented by the following command not running since it is a dry run. 
 docker push gcr.io/golden-project/dry-run-runner:prefix-current
-[XPK] Temp file (2eed164577b237a7b764f6adf46b9c249551c2200e440ef0f696c53dfb2d1cb5) content: 
+[XPK] Temp file (2c5ab381c0d643f8512a07d296d411413080ec652c15e8c676fd58435de5a327) content: 
 apiVersion: jobset.x-k8s.io/v1alpha2
 kind: JobSet
 metadata:
@@ -136,7 +136,7 @@ spec:
                   exit $EXIT_CODE
                 resources:
                   limits:
-                    google.com/tpu: 2
+                    google.com/tpu: 4
 
                 volumeMounts:
                 - mountPath: /dev/shm
@@ -156,7 +156,7 @@ spec:
               
 
 [XPK] Task: `Creating Workload` is implemented by the following command not running since it is a dry run. 
-kubectl apply -f 2eed164577b237a7b764f6adf46b9c249551c2200e440ef0f696c53dfb2d1cb5
+kubectl apply -f 2c5ab381c0d643f8512a07d296d411413080ec652c15e8c676fd58435de5a327
 [XPK] Task: `GKE Dashboard List` is implemented by the following command not running since it is a dry run. 
 gcloud monitoring dashboards list --project=golden-project --filter="displayName:'GKE - TPU Monitoring Dashboard'" --format="value(name)" --verbosity=error
 [XPK] Check statistics and outlier mode of GKE metrics here: https://console.cloud.google.com/monitoring/dashboards/builder/0?project=golden-project&f.rlabel.cluster_name.ClusterName=golden-cluster. To view the metric data for your workload, select golden-workload from the JobName filter on the dashboard.

{xpk-1.1.1 → xpk-1.1.2}/src/xpk/core/docker_container.py

@@ -181,7 +181,9 @@ def get_main_container(
             tpu_stacktrace_terminate_command=tpu_stacktrace_terminate_command,
             gpu_workload_terminate_command=gpu_workload_terminate_command,
             xpk_internal_commands=xpk_internal_commands,
-            resources=get_main_container_resources(args, system, resource_type),
+            resources=get_main_container_resources(
+                args, system, resource_type, parallel_containers
+            ),
             volume_mounts=volume_mounts,
         )
     )

{xpk-1.1.1 → xpk-1.1.2}/src/xpk/core/docker_resources.py

@@ -23,7 +23,10 @@ from ..utils.execution_context import is_dry_run
 
 
 def get_main_container_resources(
-    args, system: SystemCharacteristics, resource_type
+    args,
+    system: SystemCharacteristics,
+    resource_type: str,
+    parallel_containers: int,
 ) -> str:
   """Resources for the main container.
   Args:
@@ -53,10 +56,7 @@ def get_main_container_resources(
     offset_vCPUs = int(system.chips_per_vm) * 0.95
     return f'{resource_type}: {offset_vCPUs}'
 
-  return (
-      f'{resource_type}:'
-      f' {int(system.chips_per_vm / system.parallel_containers)}'
-  )
+  return f'{resource_type}: {int(system.chips_per_vm / parallel_containers)}'
 
 
 def get_env_container(args, system: SystemCharacteristics) -> str:

{xpk-1.1.1 → xpk-1.1.2/src/xpk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xpk
-Version: 1.1.1
+Version: 1.1.2
 Summary: xpk helps Cloud developers to orchestrate training jobs on accelerators on GKE.
 Author-email: XPK team <xpk-code-reviewers@google.com>
 License: Apache-2.0