-
Notifications
You must be signed in to change notification settings - Fork 3.7k
Pick up vector length from 'zvlXXXb' (RVV) mattr for riscv #17641
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
4a1912a to
2b186e9
Compare
| name="C", | ||
| ) | ||
| s = te.create_schedule(C.op) | ||
| s[C].vectorize(s[C].op.axis[0]) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
we are moving away from te schedule very soon, so let us instead use tir schedule
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Reworked, it is even more simple now, I was expecting relay to go, not realizing the same about te.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
te compute will stay, but we wanted to remove te.schedule
8a49899 to
146157d
Compare
146157d to
4223771
Compare
|
This is ready for review. |
|
@tvm-bot rerun |
|
Failed to re-run CI in https://github.com/apache/tvm/actions/runs/13421658268 with response |
e19c33f to
5d2671f
Compare
5d2671f to
ec8aa6b
Compare
This PR is an enhancement for #17631.
New functionality:
-mattrarguments, if user would supply such.Enhanchements:
GetVectorWidth()@LLVMTargetInfo a shared public function, reachable from python side too.This not alter any behavior except riscv targets.
This work here is a preliminary before enable TVM's SVE for riscv too.