Skip to content

Commit

Permalink
dyn_quan group size test
Browse files Browse the repository at this point in the history
  • Loading branch information
isanghao committed Nov 5, 2024
1 parent 5c94cf6 commit 4b87ec6
Showing 1 changed file with 2 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -376,6 +376,7 @@ INSTANTIATE_TEST_SUITE_P(MatMulCompressedWeights_corner_cases_big,
// per_tensor_zp=0 is not supported
// transpose_weights is not supported
// weight precision u4 is only supported
const std::vector<uint64_t> group_size = {32, 128, UINT64_MAX};
INSTANTIATE_TEST_SUITE_P(smoke_MatMulCompressedWeights_dyn_quan,
MatmulWeightsDecompression,
::testing::Combine(::testing::Values(ShapeParams{{{-1, -1, 4096}, {{1, 1, 4096}}}, {1, 4096, 4096}}), // shape
Expand All @@ -385,7 +386,7 @@ INSTANTIATE_TEST_SUITE_P(smoke_MatMulCompressedWeights_dyn_quan,
::testing::ValuesIn(add_decompression_sub),
::testing::Values(true),
::testing::Values(true), // per_tensor_zp
::testing::Values(UINT64_MAX)),
::testing::ValuesIn(group_size)),
MatmulWeightsDecompression::get_test_case_name);

} // namespace

0 comments on commit 4b87ec6

Please sign in to comment.