@ -696,7 +696,7 @@ bool TestAllConv2d(
|
||||
return false;
|
||||
}
|
||||
|
||||
// If CUTLASS_UNIT_TEST_PROBLEM_COUNT is set reduce the the number of tested problem counts
|
||||
// If CUTLASS_UNIT_TEST_PROBLEM_COUNT is set reduce the number of tested problem counts
|
||||
if (CutlassUnitTestProblemCount() &&
|
||||
testbed.tested_problem_count > CutlassUnitTestProblemCount()) {
|
||||
return true;
|
||||
@ -742,7 +742,7 @@ bool TestAllConv2d(
|
||||
}
|
||||
// Sweep split-k-slice using serial and prallel reduction with non-unity alpha and non-zero beta for
|
||||
// a single conv2d problem size. Convolution unit tests take a long time to run so only sweep parameters
|
||||
// which are abolutely neccessary to catch functional bugs. The below code does provide option to sweep
|
||||
// which are abolutely necessary to catch functional bugs. The below code does provide option to sweep
|
||||
// alpha and beta for local testing, but only runs one value for alpha and beta.
|
||||
cutlass::conv::Conv2dProblemSize conv2d_split_k_test_size (
|
||||
{1, 17, 11, 288}, // input size (NHWC)
|
||||
@ -784,7 +784,7 @@ bool TestAllConv2d(
|
||||
return false;
|
||||
}
|
||||
|
||||
// If CUTLASS_UNIT_TEST_PROBLEM_COUNT is set reduce the the number of tested problem counts
|
||||
// If CUTLASS_UNIT_TEST_PROBLEM_COUNT is set reduce the number of tested problem counts
|
||||
if (CutlassUnitTestProblemCount() &&
|
||||
testbed.tested_problem_count > CutlassUnitTestProblemCount()) {
|
||||
return true;
|
||||
|
||||
Reference in New Issue
Block a user