From d9295250396eafe7d16c8730104f669cb66777e3 Mon Sep 17 00:00:00 2001 From: Cedric Nugteren Date: Mon, 31 Dec 2018 18:49:12 +0100 Subject: Added support for the convgemm tuner in the tuner database --- scripts/database/database/clblast.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'scripts/database') diff --git a/scripts/database/database/clblast.py b/scripts/database/database/clblast.py index 86444f68..ce76f305 100644 --- a/scripts/database/database/clblast.py +++ b/scripts/database/database/clblast.py @@ -24,7 +24,9 @@ DEVICE_ATTRIBUTES = ["clblast_device_name", "clblast_device_architecture", "device_core_clock", "device_compute_units"] KERNEL_ATTRIBUTES = ["precision", "kernel_family"] ARGUMENT_ATTRIBUTES = ["arg_m", "arg_n", "arg_k", "arg_alpha", "arg_beta", - "arg_from", "arg_to", "arg_step"] + "arg_from", "arg_to", "arg_step", + "arg_channels", "arg_height", "arg_width", "arg_kernel_h", "arg_kernel_w", + "arg_num_kernels", "arg_batch_count"] ATTRIBUTES = DEVICE_ATTRIBUTES + DEVICE_TYPE_ATTRIBUTES + KERNEL_ATTRIBUTES + ARGUMENT_ATTRIBUTES GROUP_ATTRIBUTES = DEVICE_TYPE_ATTRIBUTES + KERNEL_ATTRIBUTES + ["kernel"] + ARGUMENT_ATTRIBUTES -- cgit v1.2.3