From 776cb95c964943947445bbb9ad9df0e2781274b8 Mon Sep 17 00:00:00 2001 From: Madiator2011 Date: Tue, 6 Aug 2024 13:03:41 +0200 Subject: [PATCH] Update docker-bake.hcl --- official-templates/better-base/docker-bake.hcl | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/official-templates/better-base/docker-bake.hcl b/official-templates/better-base/docker-bake.hcl index 9de52e6..b9dea6c 100644 --- a/official-templates/better-base/docker-bake.hcl +++ b/official-templates/better-base/docker-bake.hcl @@ -14,6 +14,17 @@ group "default" { ] } +group "rocm" { + targets = [ + "py310-rocm56", + "py310-rocm57", + "py310-rocm602", + "py310-rocm60", + "py310-rocm61", + "py310-rocm612" + ] +} + target "py311-cuda121" { contexts = { scripts = "../../container-template"