Skip to content

Commit 0c00d40

Browse files
committed
enable cutedsl's PDL
Signed-off-by: Zhenhuan Chen <chenzhh3671@gmail.com>
1 parent 6bccff7 commit 0c00d40

File tree

1 file changed

+1
-1
lines changed
  • tensorrt_llm/_torch/cute_dsl_kernels/blackwell

1 file changed

+1
-1
lines changed

tensorrt_llm/_torch/cute_dsl_kernels/blackwell/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@
5555
from cutlass.cute.typing import AddressSpace, Numeric, Pointer, Type
5656
from cutlass.cutlass_dsl import T, dsl_user_op
5757

58-
TRTLLM_ENABLE_PDL = os.environ.get("TRTLLM_ENABLE_PDL", "0") == "1"
58+
TRTLLM_ENABLE_PDL = os.environ.get("TRTLLM_ENABLE_PDL", "1") == "1"
5959

6060

6161
# WAR for CuTeDSL make_ptr implementation

0 commit comments

Comments
 (0)