| [AMD] Support ExtractSliceOp for AxisInfo (#7094)
This commit updates AxisInfo to support backend callbacks to
enable recognizing backend ops.
One can use ExtractSliceOp to slice tensors of pointers to refine
tt.load or tt.store. The TritonAMDGPUConvertToBufferOpsBase
will fail to perform negativity analysis due to the presence of
ExtractSliceOp which after rewrites is going to slice tensors of
offsets. This PR addresses the issue. | 11 个月前 |