saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-26 19:52:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[CUDA] Remove shape warnings in NHWC <> NCHW unit tests (#17992)

Browse source 
There were some warning in https://dev.azure.com/onnxruntime/onnxruntime/_build/results?buildId=1170770
e.g.
```
[ RUN      ] CudaNhwcTypedTest/1.AveragePoolNhwcPad @ /home/administrator/onnxruntime/onnxruntime/test/providers/cuda/nhwc/pool_test.cc:84
[W:onnxruntime:Default, graph.cc:108 MergeShapeInfo] Error merging shape info for output. 'Y' source:{1,16,66,66} target:{1,16,67,67}. Falling back to lenient merge.
```
These warnings where not specific to NHWC or NCHW but were just a miscalculation of output shape in some tests.
This commit is contained in:
Maximilian Müller 2023-10-27 01:32:01 +02:00 committed by GitHub
parent 7c18c60bc2
commit b7bee621cd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
onnxruntime/test/providers/cuda/nhwc/conv_transpose_test.cc
View file

@ -42,12 +42,16 @@ struct ConvTransposeOp {
test->AddAttribute("pads", padding);
if (!output_padding.empty()) {
test->AddAttribute("output_padding", output_padding);
} else {
output_padding = {0, 0, 0, 0};
}
std::vector<int64_t> output_dims = {
input_dims[0], channels,
(kernel_shape[1] - 1) * dilations[1] + (input_dims[2] - 1) * strides[1] - (padding[1] + padding[0]) + 1,
(kernel_shape[0] - 1) * dilations[0] + (input_dims[3] - 1) * strides[0] - (padding[3] + padding[2]) + 1};
(kernel_shape[1] - 1) * dilations[1] + (input_dims[2] - 1) * strides[1] - (padding[1] + padding[0]) + 1 +
output_padding[2],
(kernel_shape[0] - 1) * dilations[0] + (input_dims[3] - 1) * strides[0] - (padding[3] + padding[2]) + 1 +
output_padding[3]};
std::vector<T> output_data = FillZeros<T>(output_dims);
test->AddOutput<T>("Y", output_dims, output_data);

4
onnxruntime/test/providers/cuda/nhwc/pool_test.cc
View file

@ -31,8 +31,8 @@ struct PoolOp {
std::vector<int64_t> output_dims = {
input_dims[0], channels,
(kernel_shape[1] - 1) + (input_dims[2] - 1) * strides[1] - (padding[1] + padding[0]) + 1,
(kernel_shape[0] - 1) + (input_dims[3] - 1) * strides[0] - (padding[3] + padding[2]) + 1};
(input_dims[2] - (kernel_shape[0] - 1) + padding[1] + padding[0] - 1) / strides[0] + 1,
(input_dims[3] - (kernel_shape[1] - 1) + padding[3] + padding[2] - 1) / strides[1] + 1};
std::vector<T> output_data = FillZeros<T>(output_dims);
test->AddOutput<T>("Y", output_dims, output_data);