diff options
| author | Cedric Nugteren <web@cedricnugteren.nl> | 2018-12-31 18:49:12 +0100 |
|---|---|---|
| committer | Cedric Nugteren <web@cedricnugteren.nl> | 2018-12-31 18:49:12 +0100 |
| commit | d9295250396eafe7d16c8730104f669cb66777e3 (patch) | |
| tree | 5d4dc8eb0e1b78fcee0c62938607bee740c74a60 /scripts | |
| parent | 153ac06cf262d2680d0152933156b1d1e15b3f86 (diff) | |
Added support for the convgemm tuner in the tuner database
Diffstat (limited to 'scripts')
| -rw-r--r-- | scripts/database/database/clblast.py | 4 |
1 files changed, 3 insertions, 1 deletions
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 |