Skip to content

Commit 8acba94

Browse files
committed
Update on "[Ez] Enable Vulkan 4-bit weight only quantization in export_llama"
As title. Differential Revision: [D64406456](https://our.internmc.facebook.com/intern/diff/D64406456/) [ghstack-poisoned]
2 parents 995afe5 + 55a743b commit 8acba94

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

backends/vulkan/test/op_tests/linear_weight_int4_test.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -202,15 +202,15 @@ void test_vulkan_linear_int4(
202202
ASSERT_TRUE(at::allclose(vk_out, out_ref, 1e-4, 1e-4));
203203
}
204204

205-
TEST(VulkanSDPATest, test_reference_impl) {
205+
TEST(VulkanInt4LinearTest, test_reference_impl) {
206206
test_reference_linear_int4(
207207
/*B = */ 1,
208208
/*M = */ 4,
209209
/*K = */ 128,
210210
/*N = */ 32);
211211
}
212212

213-
TEST(VulkanSDPATest, test_vulkan_impl) {
213+
TEST(VulkanInt4LinearTest, test_vulkan_impl) {
214214
if (!vkcompute::api::context()
215215
->adapter_ptr()
216216
->has_full_int8_buffers_support()) {

0 commit comments

Comments
 (0)