Skip to content

Commit 460b76b

Browse files
committed
int8 reference tests
Signed-off-by: Iqbal Saraf <[email protected]>
1 parent 2f0f780 commit 460b76b

File tree

6 files changed

+10
-5
lines changed

6 files changed

+10
-5
lines changed

tests/aiu_addons/conftest.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -74,11 +74,16 @@ def get_gptq_gemm_inputs(request) -> tuple[torch.Tensor, ...]:
7474
"atype": "per_tensor_symm", # per_tensor_asymm, per_token
7575
"smoothquant": False,
7676
},
77-
# {
78-
# "wtype": "per_channel", # per_channel
79-
# "atype": "per_tensor_symm", # per_tensor_asymm, per_token
80-
# "smoothquant": False,
81-
# },
77+
{
78+
"wtype": "per_tensor", # per_channel
79+
"atype": "per_tensor_asymm", # per_tensor_asymm, per_token
80+
"smoothquant": False,
81+
},
82+
{
83+
"wtype": "per_channel", # per_channel
84+
"atype": "per_tensor_symm", # per_tensor_asymm, per_token
85+
"smoothquant": False,
86+
},
8287
]
8388

8489

Binary file not shown.
Binary file not shown.
3.06 KB
Binary file not shown.
Binary file not shown.
3.05 KB
Binary file not shown.

0 commit comments

Comments
 (0)