Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
214 changes: 115 additions & 99 deletions backends/vulkan/test/custom_ops/q8ta_q8csw_q8to_conv2d.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -164,7 +164,8 @@ TestCase create_test_case_from_config(

test_case.add_output_spec(output);

test_case.set_abs_tolerance(output_scale_val + 1e-4f);
test_case.set_abs_tolerance(output_scale_val + 2e-3f);
test_case.set_rel_tolerance(1e-2);

return test_case;
}
Expand Down Expand Up @@ -206,115 +207,130 @@ std::vector<TestCase> generate_quantized_conv2d_test_cases() {
std::vector<TestCase> test_cases;

std::vector<Conv2dConfig> configs = {
// Pointwise convolutions: kernel size 1x1
{OutInChannels(32, 3),
InputSize2D(64, 64),
KernelSize(1, 1),
Stride(1, 1),
Padding(0, 0),
Dilation(1, 1),
1},
{OutInChannels(64, 32),
InputSize2D(32, 32),
KernelSize(1, 1),
Stride(1, 1),
Padding(0, 0),
Dilation(1, 1),
1},
{OutInChannels(96, 64),
InputSize2D(16, 16),
KernelSize(1, 1),
Stride(1, 1),
Padding(0, 0),
Dilation(1, 1),
1},
{OutInChannels(13, 7),
InputSize2D(57, 33),
KernelSize(1, 1),
Stride(1, 1),
Padding(0, 0),
Dilation(1, 1),
1},
// General 2D convolutions
{OutInChannels(32, 3),
InputSize2D(64, 64),
KernelSize(3, 3),
Stride(1, 1),
Padding(1, 1),
Dilation(1, 1),
1},
{OutInChannels(32, 3),
InputSize2D(64, 64),
KernelSize(3, 3),
Stride(2, 2),
Padding(1, 1),
Dilation(1, 1),
1},
{OutInChannels(64, 32),
InputSize2D(8, 8),
KernelSize(3, 3),
Stride(1, 1),
Padding(1, 1),
Dilation(1, 1),
1},
{OutInChannels(64, 32),
InputSize2D(64, 64),
KernelSize(3, 3),
Stride(1, 1),
Padding(1, 1),
Dilation(1, 1),
1},
{OutInChannels(64, 32),
InputSize2D(64, 64),
KernelSize(3, 3),
Stride(2, 2),
Padding(1, 1),
Dilation(1, 1),
1},
{OutInChannels(16, 32),
InputSize2D(77, 77),
KernelSize(3, 3),
Stride(1, 1),
Padding(1, 1),
Dilation(1, 1),
1},
// Grouped convolutions
{OutInChannels(64, 32),
InputSize2D(64, 64),
// // Pointwise convolutions: kernel size 1x1
// {OutInChannels(32, 3),
// InputSize2D(64, 64),
// KernelSize(1, 1),
// Stride(1, 1),
// Padding(0, 0),
// Dilation(1, 1),
// 1},
// {OutInChannels(64, 32),
// InputSize2D(32, 32),
// KernelSize(1, 1),
// Stride(1, 1),
// Padding(0, 0),
// Dilation(1, 1),
// 1},
// {OutInChannels(96, 64),
// InputSize2D(16, 16),
// KernelSize(1, 1),
// Stride(1, 1),
// Padding(0, 0),
// Dilation(1, 1),
// 1},
// {OutInChannels(13, 7),
// InputSize2D(57, 33),
// KernelSize(1, 1),
// Stride(1, 1),
// Padding(0, 0),
// Dilation(1, 1),
// 1},
// // General 2D convolutions
// {OutInChannels(32, 3),
// InputSize2D(64, 64),
// KernelSize(3, 3),
// Stride(1, 1),
// Padding(1, 1),
// Dilation(1, 1),
// 1},
// {OutInChannels(32, 3),
// InputSize2D(64, 64),
// KernelSize(3, 3),
// Stride(2, 2),
// Padding(1, 1),
// Dilation(1, 1),
// 1},
// {OutInChannels(64, 32),
// InputSize2D(8, 8),
// KernelSize(3, 3),
// Stride(1, 1),
// Padding(1, 1),
// Dilation(1, 1),
// 1},
// {OutInChannels(64, 32),
// InputSize2D(64, 64),
// KernelSize(3, 3),
// Stride(1, 1),
// Padding(1, 1),
// Dilation(1, 1),
// 1},
// {OutInChannels(64, 32),
// InputSize2D(64, 64),
// KernelSize(3, 3),
// Stride(2, 2),
// Padding(1, 1),
// Dilation(1, 1),
// 1},
// {OutInChannels(16, 32),
// InputSize2D(77, 77),
// KernelSize(3, 3),
// Stride(1, 1),
// Padding(1, 1),
// Dilation(1, 1),
// 1},
// // Grouped convolutions
// {OutInChannels(64, 32),
// InputSize2D(64, 64),
// KernelSize(3, 3),
// Stride(1, 1),
// Padding(1, 1),
// Dilation(1, 1),
// 2},
// {OutInChannels(96, 96),
// InputSize2D(81, 81),
// KernelSize(3, 3),
// Stride(2, 2),
// Padding(1, 1),
// Dilation(1, 1),
// 3},
// {OutInChannels(96, 96),
// InputSize2D(64, 64),
// KernelSize(5, 5),
// Stride(2, 2),
// Padding(2, 2),
// Dilation(1, 1),
// 4},
// // Performance cases (pointwise)
// {OutInChannels(128, 128),
// InputSize2D(128, 128),
// KernelSize(1, 1),
// Stride(1, 1),
// Padding(0, 0),
// Dilation(1, 1),
// 1},
// {OutInChannels(128, 128),
// InputSize2D(128, 128),
// KernelSize(1, 1),
// Stride(1, 1),
// Padding(0, 0),
// Dilation(1, 1),
// 1},
// Performance cases (grouped convs)
{OutInChannels(64, 64),
InputSize2D(128, 128),
KernelSize(3, 3),
Stride(1, 1),
Padding(1, 1),
Dilation(1, 1),
2},
{OutInChannels(96, 96),
InputSize2D(81, 81),
InputSize2D(128, 128),
KernelSize(3, 3),
Stride(2, 2),
Padding(1, 1),
Dilation(1, 1),
3},
{OutInChannels(96, 96),
InputSize2D(64, 64),
KernelSize(5, 5),
Stride(2, 2),
Padding(2, 2),
Dilation(1, 1),
4},
// Performance cases (pointwise)
{OutInChannels(128, 128),
InputSize2D(128, 128),
KernelSize(1, 1),
Stride(1, 1),
Padding(0, 0),
Dilation(1, 1),
1},
{OutInChannels(128, 128),
InputSize2D(128, 128),
KernelSize(1, 1),
Stride(1, 1),
Padding(0, 0),
Dilation(1, 1),
1},
// Performance cases (general 2d convs)
{OutInChannels(32, 3),
InputSize2D(256, 256),
Expand Down
Loading