File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -1749,7 +1749,7 @@ GemmDescriptor CreateGemmDescriptorConvFwd(const conv::ProblemDescription& probl
17491749 yDesc.GetLengths () | std::views::drop (2 ) | std::views::take (yDesc.GetLengths ().size () - 2 );
17501750
17511751 bool isColMajor = problem.IsLayoutNHWC ();
1752- bool transA = problem. IsLayoutNHWC () ;
1752+ bool transA = isColMajor ;
17531753 bool transB = problem.IsLayoutNHWC () ? false : (wDesc.GetType () == miopenInt8);
17541754 int m = wei_k;
17551755 int n = static_cast <int >(std::accumulate (
@@ -2193,7 +2193,7 @@ GemmDescriptor CreateGemmDescriptorGroupConvFwd(const conv::ProblemDescription&
21932193 yDesc.GetLengths () | std::views::drop (2 ) | std::views::take (yDesc.GetLengths ().size () - 2 );
21942194
21952195 bool isColMajor = problem.IsLayoutNHWC ();
2196- bool transA = problem. IsLayoutNHWC () ;
2196+ bool transA = isColMajor ;
21972197 bool transB = false ;
21982198 int m = wei_k / groupCount;
21992199 int n = static_cast <int >(std::accumulate (
Original file line number Diff line number Diff line change @@ -43,7 +43,7 @@ float Col2ImGPU(const Handle& handle,
4343 std::size_t im_offset,
4444 miopenDataType_t type,
4545 bool layoutNHWC,
46- const int num_groups);
46+ int num_groups);
4747
4848float Col2Im3dGPUBatched (const Handle& handle,
4949 ConstData_t col,
You can’t perform that action at this time.
0 commit comments