fix docker commands

Ruturaj4 · Ruturaj4 · commit 7483227bbae4 · 2026-01-14T17:19:08.000-06:00
diff --git a/jax_rocm_plugin/build/rocm/ci_build b/jax_rocm_plugin/build/rocm/ci_build
@@ -141,9 +141,10 @@ def dist_wheels(
     # NOTE(mrodden): bazel times out without --init, probably blocking on a zombie PID
     cmd.extend(
         [
-            "-it",
             "--init",
             "--rm",
+            "--shm-size",
+            "64G",
             "-e",
             "GIT_DIR=/repo/.git",
             "-e",
@@ -154,8 +155,6 @@ def dist_wheels(
             "bash",
             "-c",
             container_cmd,
-            "--shm-size",
-            "64G",
         ]
     )
 

Original file line number	Diff line number	Diff line change
`@@ -141,9 +141,10 @@ def dist_wheels(`
`141`	`141`	`# NOTE(mrodden): bazel times out without --init, probably blocking on a zombie PID`
`142`	`142`	`cmd.extend(`
`143`	`143`	`[`
`144`		`- "-it",`
`145`	`144`	`"--init",`
`146`	`145`	`"--rm",`
	`146`	`+ "--shm-size",`
	`147`	`+ "64G",`
`147`	`148`	`"-e",`
`148`	`149`	`"GIT_DIR=/repo/.git",`
`149`	`150`	`"-e",`
`@@ -154,8 +155,6 @@ def dist_wheels(`
`154`	`155`	`"bash",`
`155`	`156`	`"-c",`
`156`	`157`	`container_cmd,`
`157`		`- "--shm-size",`
`158`		`- "64G",`
`159`	`158`	`]`
`160`	`159`	`)`
`161`	`160`