| wrap_triton
Co-authored-by: 1479518308<cuiduo1@huawei.com>
# message auto-generated for no-merge-commit merge:
!26741 merge master into master
wrap_triton
Created-by: cuiduo
Commit-by: 1479518308
Merged-by: ascend-robot
Description: <!-- Thanks for sending a pull request!
-->
**What type of PR is this?**
> /kind task
**What does this PR do / why do we need it**:Add wrap_triton function adaptation. Add triton operator adaptation modules in the files torch._inductor.lowering.py and torch._inductor.lowering_fx.py. Add triton_kernel_wrapper_mutation to the whitelist in the file torch._inductor.lowering_op_list.py.
**Which issue(s) this PR fixes**:
<!--
*Automatically closes linked issue when PR is merged.
Usage: Fixes #<issue number>, or Fixes (paste link of issue).
-->
Fixes #Fixes the issue where custom Triton operators under the wrap_triton function fail to function as expected.
See merge request: Ascend/pytorch!26741 | 6 个月前 |
