search

intel / torch-xpu-ops

Apache License 2.0
30 stars 21 forks source link

Add c shim wrapper codegen. #1021

Closed etaf closed 1 week ago

etaf commented 3 weeks ago

Add c shim wrapper codegen. Support generate c shim wrapper for OPs in torch_xpu_ops which extend the c shim wrapper of Pytorch in-tree XPU ops.

etaf commented 3 weeks ago

This PR is relied upon by AOT Inductor PR https://github.com/pytorch/pytorch/pull/139026. And should be able to pass the CI after Pytorch PR https://github.com/pytorch/pytorch/pull/136742 landed. The design is descripted in https://github.com/pytorch/pytorch/pull/136742

EikanWang commented 3 weeks ago

CI failure. Because it depends on PyTorch PR, right?

etaf commented 1 week ago

@fengyuan14 can I merge this PR?