From 7498c4512b86f4f74782e60d387bfd670d503888 Mon Sep 17 00:00:00 2001 From: w4yne Date: Thu, 10 Oct 2024 11:35:18 +0800 Subject: [PATCH] [KUNLUNXIN] kernel timing bugfix --- operation/vendors/kunlunxin/xpytorch029/Dockerfile | 1 + 1 file changed, 1 insertion(+) diff --git a/operation/vendors/kunlunxin/xpytorch029/Dockerfile b/operation/vendors/kunlunxin/xpytorch029/Dockerfile index 7ecbad8f3..02ae43566 100644 --- a/operation/vendors/kunlunxin/xpytorch029/Dockerfile +++ b/operation/vendors/kunlunxin/xpytorch029/Dockerfile @@ -17,3 +17,4 @@ RUN pip3 install munch ENV TRITON_XPU_ARCH 3 ENV CUDART_DUMMY_REGISTER 1 +ENV XPURT_DISPATCH_MODE PROFILING