Changed the override-parameters test such that it is compatible with more devices

1 files changed, 8 insertions, 4 deletions

diff --git a/test/correctness/misc/override_parameters.cpp b/test/correctness/misc/override_parameters.cpp
index 54d18cfa..a4cecf0d 100644
--- a/test/correctness/misc/override_parameters.cpp
+++ b/test/correctness/misc/override_parameters.cpp
@@ -43,9 +43,9 @@ size_t RunOverrideTests(int argc, char *argv[], const bool silent, const std::st
   const auto platform_id = GetArgument(arguments, help, kArgPlatform, ConvertArgument(std::getenv("CLBLAST_PLATFORM"), size_t{0}));
   const auto device_id = GetArgument(arguments, help, kArgDevice, ConvertArgument(std::getenv("CLBLAST_DEVICE"), size_t{0}));
   auto args = Arguments<T>{};
-  args.m = GetArgument(arguments, help, kArgM, size_t{512});
-  args.n = GetArgument(arguments, help, kArgN, size_t{512});
-  args.k = GetArgument(arguments, help, kArgK, size_t{512});
+  args.m = GetArgument(arguments, help, kArgM, size_t{256});
+  args.n = GetArgument(arguments, help, kArgN, size_t{256});
+  args.k = GetArgument(arguments, help, kArgK, size_t{256});
   args.a_ld = GetArgument(arguments, help, kArgALeadDim, args.k);
   args.b_ld = GetArgument(arguments, help, kArgBLeadDim, args.n);
   args.c_ld = GetArgument(arguments, help, kArgCLeadDim, args.n);
@@ -124,11 +124,15 @@ size_t RunOverrideTests(int argc, char *argv[], const bool silent, const std::st
 // =================================================================================================
 } // namespace clblast
 
+// Shortcuts to the clblast namespace
+using float2 = clblast::float2;
+using double2 = clblast::double2;
+
 // Main function (not within the clblast namespace)
 int main(int argc, char *argv[]) {
   auto errors = size_t{0};
   errors += clblast::RunOverrideTests<float>(argc, argv, false, "SGEMM");
-  errors += clblast::RunOverrideTests<double>(argc, argv, true, "DGEMM");
+  errors += clblast::RunOverrideTests<float2>(argc, argv, true, "CGEMM");
   if (errors > 0) { return 1; } else { return 0; }
 }