File tree Expand file tree Collapse file tree 3 files changed +9
-17
lines changed Expand file tree Collapse file tree 3 files changed +9
-17
lines changed Original file line number Diff line number Diff 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
4441const static std::vector<std::vector<size_t >> stride_range = {{1 }};
Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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 ;
You can’t perform that action at this time.
0 commit comments