haraldwolff/CLBlast
1
0
Fork 0
mirror of https://github.com/CNugteren/CLBlast.git synced 2024-07-07 12:23:46 +02:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse source
This commit is contained in:
Cedric Nugteren 2017-02-18 11:22:07 +01:00
parent 2e0951c6dc
commit 7b2170818f
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
test/correctness/misc/override_parameters.cpp
View file

@ -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; }
}