diff options
author | Young Jin Kim <youki@microsoft.com> | 2019-07-04 01:00:04 +0300 |
---|---|---|
committer | Young Jin Kim <youki@microsoft.com> | 2019-07-04 01:00:04 +0300 |
commit | 843c84ad717d8e84b47b8ce5549cd13e0b13c6f1 (patch) | |
tree | 67935f5f678f21602428b7a284ac279fc64b580c /src/graph | |
parent | 1b03ed688b5a9e7bd3db7ddb86f9b02efcc8a8b8 (diff) |
Fix a bug for GPU inference
Diffstat (limited to 'src/graph')
-rwxr-xr-x | src/graph/expression_operators.cpp | 2 |
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. |