🚀 The feature, motivation and pitch
So that it has a similar UX as torch.empty.
Could simplify auto-generated codes, like those by Inductor.
For eager codes, explicit MemPool would work better for y = foo(x) where y is desired to be in symm mem.
Alternatives
No response
Additional context
No response
cc @ptrblck @msaroufim @eqy @jerryzh168 @tinglvv @nWEIdia
🚀 The feature, motivation and pitch
So that it has a similar UX as
torch.empty.Could simplify auto-generated codes, like those by Inductor.
For eager codes, explicit MemPool would work better for
y = foo(x)where y is desired to be in symm mem.Alternatives
No response
Additional context
No response
cc @ptrblck @msaroufim @eqy @jerryzh168 @tinglvv @nWEIdia