Skip to content

Conversation

@xinxilwl
Copy link
Contributor

@xinxilwl xinxilwl commented Apr 23, 2025

I fixed the bug below:
17587 fails on tvm.ir.assert_structural_equal in tests/python/relax/test_transform_legalize_ops_nn.py

@yongwww
Copy link
Member

yongwww commented Apr 27, 2025

cc: @MasterJH5574

@tqchen
Copy link
Member

tqchen commented May 8, 2025

@cyx-6 can you also help to cross check this

@tqchen
Copy link
Member

tqchen commented Jun 2, 2025

sorry for the delayed note, i take a closer look. In principle we should not pass Expr as part of Attrs and always treat Attrs as constant, as a result, we cannot take this change as it is. We can however, require the attention funciton to optionally takes a bias argument, and the printer should be updated to be able to print out the bias or remove bias as needed.

So the correct fix here would be to fix the printer and deduction function of attention op, if we allow optional bias in the attention operator

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants