diff --git a/docker-compose.yaml b/docker-compose.yaml
index 7d98fe4..1fd3573 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -102,6 +102,7 @@ services:
       - GPU_ARCH=rocm7.1
       - PYTORCH_TUNABLEOP_ENABLED=1
       - MIOPEN_FIND_MODE=FAST
+      - TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1
     ports:
       - "8188:8188"
     networks: