Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/marian-nmt/marian.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
path: root/src/graph
diff options
context:
space:
mode:
authorYoung Jin Kim <youki@microsoft.com>2019-07-04 01:00:04 +0300
committerYoung Jin Kim <youki@microsoft.com>2019-07-04 01:00:04 +0300
commit843c84ad717d8e84b47b8ce5549cd13e0b13c6f1 (patch)
tree67935f5f678f21602428b7a284ac279fc64b580c /src/graph
parent1b03ed688b5a9e7bd3db7ddb86f9b02efcc8a8b8 (diff)
Fix a bug for GPU inference
Diffstat (limited to 'src/graph')
-rwxr-xr-xsrc/graph/expression_operators.cpp2
1 files changed, 1 insertions, 1 deletions
diff --git a/src/graph/expression_operators.cpp b/src/graph/expression_operators.cpp
index 6b0ec472..a7bccaab 100755
--- a/src/graph/expression_operators.cpp
+++ b/src/graph/expression_operators.cpp
@@ -411,7 +411,7 @@ Expr affine(Expr a, Expr b, Expr bias, bool transA, bool transB, float scale) {
float clipValue = a->graph()->getBackend()->getClip();
- if(a->graph()->getBackend()->isOptimized() && device == DeviceType::cpu) {
+ if(device == DeviceType::cpu && a->graph()->getBackend()->isOptimized()) {
GemmType gemmType = a->graph()->getBackend()->getGemmType();
// When gemmType is set to 'auto', an autotuner decides the best algorithm available.
// A new autotuner is created, then different kinds of algorithms are added to the autotuner.