From 5d235c29402a70ab61e3d23d44d107068af233b5 Mon Sep 17 00:00:00 2001
From: lubaihua33 <56374758+lubaihua33@users.noreply.github.com>
Date: Tue, 19 Mar 2024 22:05:04 +0800
Subject: [PATCH] Add more roles related with gallery image for AITL (#3212)

---
 microsoft/utils/setup_aitl.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/microsoft/utils/setup_aitl.py b/microsoft/utils/setup_aitl.py
index 4083fdaeeb..1a79c6f9f8 100644
--- a/microsoft/utils/setup_aitl.py
+++ b/microsoft/utils/setup_aitl.py
@@ -220,8 +220,13 @@ def _set_target_role_parameters(
         "Microsoft.Compute/disks/delete",
         "Microsoft.Compute/images/read",
         "Microsoft.Compute/images/write",
+        # for testing ARM64 VHD and gallery image
         "Microsoft.Compute/galleries/images/read",
         "Microsoft.Compute/galleries/images/write",
+        "Microsoft.Compute/galleries/images/versions/read",
+        "Microsoft.Compute/galleries/images/versions/write",
+        "Microsoft.Compute/galleries/read",
+        "Microsoft.Compute/galleries/write",
         # for test VM extension running
         "Microsoft.Compute/virtualMachines/extensions/read",
         "Microsoft.Compute/virtualMachines/extensions/write",