tlc-pack / relax

Apache License 2.0
193 stars 58 forks source link

[BYOC] Minor clean up in cutlass codegen #426

Closed masahi closed 1 year ago

masahi commented 1 year ago

This was extracted from https://github.com/tlc-pack/relax/pull/417

Finding the anchor op is required only if the attributes of the anchor op are needed for codegen. In cutlass BYOC, all attributes are annotated by the python side, so dense_call and others are only used for getting struct info. So we don't need any pattern-specific logic.

@yelite @vinx13