Skip to content

Commit 029ecb1

Browse files
njriasanmeta-codesync[bot]
authored andcommitted
[GB300] [Triton] Add a Triton API to disable the experimental API in PyTorch (#696)
Summary: Pull Request resolved: #696 Adds an API so PyTorch won't select the experimental API but that its still possible. Reviewed By: adamomainz Differential Revision: D87822825 fbshipit-source-id: bf7cee9f460ac5acb7b46d3e7bc07e4b632ff894
1 parent b4a3c23 commit 029ecb1

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

python/triton/tools/experimental_descriptor.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -80,3 +80,7 @@ def create_1d_tma_descriptor_type(ptr, dim, block_dim, dtype):
8080

8181
def create_2d_tma_descriptor_type(ptr, dim1, dim0, block_dim1, block_dim0, dtype):
8282
return TmaDescKernelParamType(ptr, [dim1, dim0], [block_dim1, block_dim0], dtype)
83+
84+
85+
def enable_in_pytorch():
86+
return False

0 commit comments

Comments
 (0)