search

mlc-ai / relax

Apache License 2.0
149 stars 75 forks source link

[BugFix][ROCM] The ROCM backend Bug Fix #262

Closed zhangxiao-stack closed 1 year ago

zhangxiao-stack commented 1 year ago

When using the ROCm backend to run MLC-LLM, the following bug was encountered and fixed. 1、Module hip should be either dso exportable or binary ser izable 2、 llvm.amdgcn.ds.bpermute Intrinsic has incorrect type 3、addrspace(3) @shmem Global is external, but doesn't … …have external or weak linkage

yzh119 commented 1 year ago

Hi @zhangxiao-stack , thanks so much for doing this, would you mind contributing the changes to TVM upstream? We expect the updates in Relax repo are all about Relax compiler/runtime, otherwise rebasing would be hard.

tqchen commented 1 year ago

Thank you @zhangxiao-stack , please send it to apache tvm

tqchen commented 1 year ago

gentle ping @zhangxiao-stack

tqchen commented 1 year ago

This has been bought back to tvm. Thank you @zhangxiao-stack !