Pass computes value to libyolo_layer build (#10210)

This commit is contained in:
Nate Meyer
2024-03-05 07:04:13 -05:00
committed by GitHub
parent 30b68e59f2
commit 390403d957
4 changed files with 9 additions and 3 deletions

View File

@@ -10,12 +10,16 @@ variable "SLIM_BASE" {
variable "TRT_BASE" {
default = null
}
variable "COMPUTE_LEVEL" {
default = ""
}
target "_build_args" {
args = {
BASE_IMAGE = BASE_IMAGE,
SLIM_BASE = SLIM_BASE,
TRT_BASE = TRT_BASE
TRT_BASE = TRT_BASE,
COMPUTE_LEVEL = COMPUTE_LEVEL
}
platforms = ["linux/${ARCH}"]
}