SIGN IN SIGN UP

[ET-VK][ez] Allow logit linear layer to be lowered to Vulkan (#9951)

## Context

Due to poor performance of Vulkan's int4 linear operator, the final logit layer of the transformer model was not being delegated to vulkan, and was instead quantized and executed with the XNNPACK delegate.

However, with D72412950 / https://github.com/pytorch/executorch/pull/9883 decent performance can now be achieved with Vulkan/s int4 linear op. Therefore, the final logit layer can be lowered to Vulkan.

## Changes

* Remove limit from `VkInt4WeightOnlyQuantizer` that was causing it to ignore the logit layer of the transformer
* Do not apply XNNPACK partitioner and quantizer when lowering with Vulkan

Differential Revision: [D72480177](https://our.internmc.facebook.com/intern/diff/D72480177/)

cc @manuelcandales @cbilgin
S
Sicheng Stephen Jia committed
2cce2db0d25c63c9869d932cd8e4e20acb75c8e2
Parent: 1bef520
Committed by GitHub <noreply@github.com> on 4/7/2025, 10:55:29 PM