We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3addbbd commit 1f166baCopy full SHA for 1f166ba
1 file changed
vsmigx/vs_migraphx.cpp
@@ -850,7 +850,7 @@ static void VS_CC vsMIGXCreate(
850
{
851
size_t target = 1; // MIGX uses elements to measure strides
852
for (int i = static_cast<int>(ndim) - 1; i >= 0; i--) {
853
- if (strides[i] != target) {
+ if (lengths[i] > 1 && strides[i] != target) {
854
return set_error(
855
"invalid stride for NCHW, expects " +
856
std::to_string(target) +
@@ -914,7 +914,7 @@ static void VS_CC vsMIGXCreate(
914
915
916
917
918
919
920
0 commit comments