bug fix in ConvolutionEngineTests.cpp: when testing CPU, use HWC layout

LightningDev · Jan 2, 2016 · 5fc6f34 · 5fc6f34
1 parent 2752901
commit 5fc6f34
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Tests/UnitTests/MathTests/ConvolutionEngineTests.cpp b/Tests/UnitTests/MathTests/ConvolutionEngineTests.cpp
@@ -133,8 +133,8 @@ fprintf(stderr, "ConvolutionEngineTests.cpp %d\n", __LINE__);
 
         for (int deviceId : { -1, 0 })
         {
-            fprintf(stderr, "ConvolutionEngineTests.cpp %d\n", __LINE__);
-            auto fact = ConvFact::Create(deviceId, ConvFact::EngineType::Auto, ImageLayoutKind::CHW);
+fprintf(stderr, "ConvolutionEngineTests.cpp %d\n", __LINE__);
+            auto fact = ConvFact::Create(deviceId, ConvFact::EngineType::Auto, deviceId >= 0 ? ImageLayoutKind::CHW : ImageLayoutKind::HWC);
 fprintf(stderr, "ConvolutionEngineTests.cpp %d\n", __LINE__);
             auto eng = fact->CreateConvEngine(deviceId, 0);
 fprintf(stderr, "ConvolutionEngineTests.cpp %d\n", __LINE__);