mlc-ai / relax

Apache License 2.0
137 stars 69 forks source link

[Unity][Dlight] GeMV rule skip cases of "outer dim being grouped" #284

Closed MasterJH5574 closed 10 months ago

MasterJH5574 commented 10 months ago

Prior to this PR, the GeMV contains an assumption that "the grouped dimension of the largest tensor in GeMV must has the same iter type as the innermost dimension", which is enforced by an assertion.

Since AutoGPTQ quantization only supports KN GeMV layout, the decode-GeMV generated under the AutoGPTQ quantization has the pattern where the reduction dimension is not the innermost and is grouped. This pattern does not follow the assersion above and thus an assertion error will be thrown.

This PR updates the logic to skip transforming such cases, instead of asserting the assumption. One real AutoGPTQ decode-GeMV workload is added as a test case for future awareness.