nod-ai / SHARK-ModelDev

Unified compiler/runtime for interfacing with PyTorch Dynamo.
Apache License 2.0
95 stars 48 forks source link

torch.aten.resize_as #667

Open renxida opened 6 months ago

renxida commented 6 months ago

@kobeliu85 is working on aten.resize in https://github.com/nod-ai/SHARK-Turbine/issues/663

@kobeliu85 once you're done with implementing resize, it might be a good idea to tack this onto the same PR - - this seems like the same op, except instead of specifying a shape, the shape is acquired from another tensor.

vivekkhandelwal1 commented 3 months ago

@renxida @bangtianliu What's the current status of this issue?

bangtianliu commented 3 months ago

During my onboarding, I initially selected this task but was later asked to switch to a different one, specifically implementing layouts for the conv op from the codegen team targeting the AMD MI300 GPU. So you can remove me from the assignees list now.