Skip to content

Commit 2e0d632

Browse files
committed
add explicit buildbase option to use temporary container for backend builds
1 parent e57b755 commit 2e0d632

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

build.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2591,6 +2591,12 @@ def enable_all():
25912591
required=False,
25922592
help='Use specified Docker image in build as <image-name>,<full-image-name>. <image-name> can be "base", "gpu-base", or "pytorch".',
25932593
)
2594+
parser.add_argument(
2595+
"--use-buildbase",
2596+
default=False,
2597+
action="store_true",
2598+
help='Use local temporary "buildbase" Docker image as "base" image to build backends',
2599+
)
25942600

25952601
parser.add_argument(
25962602
"--enable-all",
@@ -2934,6 +2940,8 @@ def enable_all():
29342940

29352941
# Initialize map of docker images.
29362942
images = {}
2943+
if FLAGS.use_buildbase:
2944+
images["base"] = "tritonserver_buildbase"
29372945
for img in FLAGS.image:
29382946
parts = img.split(",")
29392947
fail_if(

0 commit comments

Comments
 (0)