fix: enable accelerated networking as a flag in azure vms

Signed-off-by: bthananjeyan <[email protected]>
ray-project · Oct 17, 2024 · a69267d · a69267d
1 parent d5fa9a0
commit a69267d
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/python/ray/autoscaler/_private/_azure/azure-vm-template.json b/python/ray/autoscaler/_private/_azure/azure-vm-template.json
@@ -107,6 +107,13 @@
             "metadata": {
                 "descriptions": "Subnet resource id."
             }
+        },
+        "enableAcceleratedNetworking": {
+            "type": "bool",
+            "defaultValue": false,
+            "metadata": {
+                "descriptions": "Whether to enable accelerated networking."
+            }
         }
     },
     "variables": {
@@ -148,7 +155,8 @@
                 ],
                 "networkSecurityGroup": {
                     "id": "[parameters('nsg')]"
-                }
+                },
+                "enableAcceleratedNetworking": "[parameters('enableAcceleratedNetworking')]"
             },
             "condition": "[parameters('provisionPublicIp')]"
         },
@@ -175,7 +183,8 @@
                 ],
                 "networkSecurityGroup": {
                     "id": "[parameters('nsg')]"
-                }
+                },
+                "enableAcceleratedNetworking": "[parameters('enableAcceleratedNetworking')]"
             },
             "condition": "[not(parameters('provisionPublicIp'))]"
         },