Skip to content

Commit

Permalink
vsmigx/vs_migraphx.cpp: fix support for flexible output
Browse files Browse the repository at this point in the history
  • Loading branch information
WolframRhodium committed May 28, 2024
1 parent 2b81819 commit 45ae058
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion vsmigx/vs_migraphx.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -855,7 +855,7 @@ static void VS_CC vsMIGXCreate(
if (lengths[0] != 1) {
return set_error("batch size must be 1");
}
if (lengths[1] != 1 && lengths[1] != 3 && !d->flexible_output_prop.empty()) {
if (lengths[1] != 1 && lengths[1] != 3 && d->flexible_output_prop.empty()) {
return set_error("output should have 1 or 3 channels, or enable \"flexible_output\"");
}
if (lengths[2] % tile_h != 0 && lengths[3] % tile_w != 0) {
Expand Down

0 comments on commit 45ae058

Please sign in to comment.