mirror of
https://github.com/CNugteren/CLBlast.git
synced 2024-08-26 14:57:02 +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 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}));
|
const auto device_id = GetArgument(arguments, help, kArgDevice, ConvertArgument(std::getenv("CLBLAST_DEVICE"), size_t{0}));
|
||||||
auto args = Arguments<T>{};
|
auto args = Arguments<T>{};
|
||||||
args.m = GetArgument(arguments, help, kArgM, size_t{512});
|
args.m = GetArgument(arguments, help, kArgM, size_t{256});
|
||||||
args.n = GetArgument(arguments, help, kArgN, size_t{512});
|
args.n = GetArgument(arguments, help, kArgN, size_t{256});
|
||||||
args.k = GetArgument(arguments, help, kArgK, size_t{512});
|
args.k = GetArgument(arguments, help, kArgK, size_t{256});
|
||||||
args.a_ld = GetArgument(arguments, help, kArgALeadDim, args.k);
|
args.a_ld = GetArgument(arguments, help, kArgALeadDim, args.k);
|
||||||
args.b_ld = GetArgument(arguments, help, kArgBLeadDim, args.n);
|
args.b_ld = GetArgument(arguments, help, kArgBLeadDim, args.n);
|
||||||
args.c_ld = GetArgument(arguments, help, kArgCLeadDim, 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
|
} // namespace clblast
|
||||||
|
|
||||||
|
// Shortcuts to the clblast namespace
|
||||||
|
using float2 = clblast::float2;
|
||||||
|
using double2 = clblast::double2;
|
||||||
|
|
||||||
// Main function (not within the clblast namespace)
|
// Main function (not within the clblast namespace)
|
||||||
int main(int argc, char *argv[]) {
|
int main(int argc, char *argv[]) {
|
||||||
auto errors = size_t{0};
|
auto errors = size_t{0};
|
||||||
errors += clblast::RunOverrideTests<float>(argc, argv, false, "SGEMM");
|
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; }
|
if (errors > 0) { return 1; } else { return 0; }
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue