Open mars1248 opened 2 months ago
Yea I don't think this is a GPU specified issue, we probally just need to lower this op. cc @wonjoolee95 let's just add this to our todo.
@wonjoolee95 Can you support it now?
Hey @vanbasten23, is this something we can add as an onboarding task?
I can you can see if we can use upstream decomp directly https://github.com/pytorch/pytorch/blob/c6cce976b25708e66ca2d7d7da1cd27c942e7a22/torch/_decomp/decompositions.py#L4422 similar to what I did in https://github.com/pytorch/xla/commit/5e51ca2365c2955bebdaaa62dfccbab3caa6cbda
❓ Questions and Help
I use this code
and then report this bug