Non-strict torch.export traces through torch_function and dispatcher/proxy machinery, so a raw Triton kernel launch is invisible as a stable export operator. The trace falls into Triton's runtime and the argument binder tries to specialize tensor arguments via data_ptr(), which fake/functional tensors cannot expose. Users then saw a generic data pointer failure instead of guidance for the supported Triton export path.
This keeps strict export behavior unchanged and rewrites only non-strict data_ptr failures that occur from Triton's runtime into an actionable error telling users to wrap the kernel in torch.library.triton_op and launch it through torch.library.wrap_triton or torch._library.capture_triton. A focused CUDA/Triton regression test covers the raw-kernel non-strict path, and the existing custom Triton export test covers the supported path.
Fixes
Approved by:
Community-Analysen & Experten-Meinungen 0
Verwandte Story-Cluster & Quellen (Vektor-KI)
Ähnliche Beiträge
Auch interessante Nachrichten trunk/94c0a41a16f7a65c690b28f5150c886d3a72dac6: Improve non-strict export error for raw Triton kernels (#185827)
Thematisch verwandte Begriffe: trunk94c0a41a16f7a65c690b28f5150c886d3a72dac6, Improve, nonstrict, export · 6 Treffer
Cobalt Strike Beacon Extractor
Videos werden geladen ...
Beiträge werden geladen ...
Videos werden geladen ...
Beiträge werden geladen ...
Videos werden geladen ...
Beiträge werden geladen ...
Videos werden geladen ...
Beiträge werden geladen ...
Videos werden geladen ...
SOCIAL SHARE CARD GENERATOR