Skip to content

Commit 4341e77

Browse files
Revert "Allow build_CS_3D_BLOCK_RC to also have composite sub-schemes. "
This reverts commit ba13730.
1 parent ba13730 commit 4341e77

File tree

3 files changed

+9
-17
lines changed

3 files changed

+9
-17
lines changed

clients/tests/accuracy_test_adhoc.cpp

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -36,9 +36,6 @@ std::vector<std::vector<size_t>> adhoc_sizes = {
3636
// SBRC 192 with special param
3737
{192, 192, 192},
3838
{192, 84, 84},
39-
40-
// Failure with build_CS_3D_BLOCK_RC
41-
{680, 128, 128},
4239
};
4340

4441
const static std::vector<std::vector<size_t>> stride_range = {{1}};

library/src/assignment_policy.cpp

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1103,14 +1103,6 @@ void AssignmentPolicy::PadPlan(ExecPlan& execPlan)
11031103
// SBCR plans combine higher dimensions in ways that confuse padding
11041104
if(u.node.scheme == CS_KERNEL_STOCKHAM_BLOCK_CR)
11051105
return;
1106-
// transpose kernels don't handle arbitrary strides,
1107-
// and with 4 or more lengths either choice of
1108-
// padding dim will trigger incorrect behaviour
1109-
if((u.node.scheme == CS_KERNEL_TRANSPOSE
1110-
|| u.node.scheme == CS_KERNEL_TRANSPOSE_XY_Z
1111-
|| u.node.scheme == CS_KERNEL_TRANSPOSE_Z_XY)
1112-
&& u.node.length.size() > 3)
1113-
return;
11141106
}
11151107

11161108
// Ensure that if we're forced to pad along one dimension

library/src/tree_node_3D.cpp

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -473,6 +473,13 @@ void BLOCKRC3DNode::AssignParams_internal()
473473
node->oDist = node->outStride[2] * node->length[0];
474474
break;
475475
}
476+
case CS_KERNEL_STOCKHAM:
477+
{
478+
node->outStride = node->inStride;
479+
node->oDist = node->iDist;
480+
node->AssignParams();
481+
break;
482+
}
476483
case CS_KERNEL_TRANSPOSE_XY_Z:
477484
{
478485
node->outStride.push_back(1);
@@ -490,12 +497,8 @@ void BLOCKRC3DNode::AssignParams_internal()
490497
break;
491498
}
492499
default:
493-
{
494-
node->outStride = node->inStride;
495-
node->oDist = node->iDist;
496-
node->AssignParams();
497-
break;
498-
}
500+
// build_CS_3D_BLOCK_RC should not have created any other node types
501+
throw std::runtime_error("Scheme Assertion Failed, unexpected node scheme.");
499502
}
500503
prev_outStride = node->outStride;
501504
prev_oDist = node->oDist;

0 commit comments

Comments
 (0)