search

tlc-pack / relax

Apache License 2.0
193 stars 58 forks source link

Remove Attr for alloc_storage and alloc_tensor #420

Closed yongwww closed 1 year ago

yongwww commented 1 year ago

follow-up work of #405

MasterJH5574 commented 1 year ago

I note that #407 was merged last night, and therefore now there quite a few uses of alloc_storage, alloc_tensor, kill_storage and kill_tensor in https://github.com/tlc-pack/relax/blob/relax/src/relax/transform/static_plan_block_memory.cc. Could you do a rebase to the latest relax branch and update them? Sorry for the inconvenience 🙏

yongwww commented 1 year ago

I note that #407 was merged last night, and therefore now there quite a few uses of alloc_storage, alloc_tensor, kill_storage and kill_tensor in https://github.com/tlc-pack/relax/blob/relax/src/relax/transform/static_plan_block_memory.cc. Could you do a rebase to the latest relax branch and update them? Sorry for the inconvenience 🙏

Sure, thanks for letting me know! have rebased to that and updated the code and tests accordingly. Please take another look at this pr.