Author: Wen-Heng (Jack) Chung Date: 2020-06-05T22:18:20-05:00 New Revision: fe71429c14c28ee85ff94a528c59b6a1436b33ac
URL: https://github.com/llvm/llvm-project/commit/fe71429c14c28ee85ff94a528c59b6a1436b33ac DIFF: https://github.com/llvm/llvm-project/commit/fe71429c14c28ee85ff94a528c59b6a1436b33ac.diff LOG: Fix input / weight tensor translation logic in Cpp preamble. Added: Modified: mlir/lib/Dialect/MIOpenOps/CppOutput/ConvertToMIOpenCPP.cpp Removed: ################################################################################ diff --git a/mlir/lib/Dialect/MIOpenOps/CppOutput/ConvertToMIOpenCPP.cpp b/mlir/lib/Dialect/MIOpenOps/CppOutput/ConvertToMIOpenCPP.cpp index b0b104bd137d..45bb8b3ea7ae 100644 --- a/mlir/lib/Dialect/MIOpenOps/CppOutput/ConvertToMIOpenCPP.cpp +++ b/mlir/lib/Dialect/MIOpenOps/CppOutput/ConvertToMIOpenCPP.cpp @@ -362,10 +362,10 @@ struct GridwiseConvolutionImplicitGemm_v4r4_)"; // TBD: remove these interim checks. if (tensorDescs.size() > 0) output << R"( - constexpr auto )" << tensorDescs[0] << " = InGlobalDesc{};"; + constexpr auto )" << tensorDescs[0] << " = WeiGlobalDesc{};"; if (tensorDescs.size() > 1) output << R"( - constexpr auto )" << tensorDescs[1] << " = WeiGlobalDesc{};"; + constexpr auto )" << tensorDescs[1] << " = InGlobalDesc{};"; if (tensorDescs.size() > 2) output << R"( constexpr auto )" << tensorDescs[2] << " = OutGlobalDesc{};"; _______________________________________________ llvm-branch-commits mailing list llvm-branch-commits@lists.llvm.org https://lists.llvm.org/cgi-bin/mailman/listinfo/llvm-branch-commits