guoyaol opened a new pull request, #15884: URL: https://github.com/apache/tvm/pull/15884
Add support: allow one dynamic tensor as torch.tensor's index In Stable Diffusion XL: e.g. `output = input1 [input2.argmax(dim = -1), ]` **Before:** we can only support int, slice, elipsis, None **Now:** we can have one dynamic torch.tensor in index, which allows us to represent SD XL correctly. Behaviors **aligns with torch.tensor**: as shown in test cases **Future TODO:** Support other "fancy indexing" methods in torch.tensor, (e.g. multiple tensors as index) which requires some new relax ops -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
