mirror of
https://github.com/CNugteren/CLBlast.git
synced 2024-07-16 19:33:00 +02:00
Changed the override-parameters test such that it is compatible with more devices
This commit is contained in:
parent
2e0951c6dc
commit
7b2170818f
|
@ -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; }
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in a new issue