[llvm-branch-commits] [mlir] fe71429 - Fix input / weight tensor translation logic in Cpp preamble.

[Wen-Heng Chung via llvm-branch-commits]

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{};";