diff options
| author | Young Jin Kim <youki@microsoft.com> | 2019-06-26 01:37:18 +0300 |
|---|---|---|
| committer | Young Jin Kim <youki@microsoft.com> | 2019-06-26 01:37:18 +0300 |
| commit | 457eef6f99d4268cc48f14e010027cd19044959f (patch) | |
| tree | 9e80222807816686649deeb6853344fd1ce962e7 /src/graph | |
| parent | 15cf6ba979e61086b4c8ee45a37249898bd44690 (diff) | |
Add thread_local to the thread specific variable. Add Gemm type check for the dot product.
Diffstat (limited to 'src/graph')
| -rwxr-xr-x | src/graph/expression_operators.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/graph/expression_operators.cpp b/src/graph/expression_operators.cpp index fac86472..60296c87 100755 --- a/src/graph/expression_operators.cpp +++ b/src/graph/expression_operators.cpp @@ -387,7 +387,8 @@ Expr dot(Expr a, Expr b, bool transA, bool transB, float scale) { // Currently only true when command line options // --optimize --cpu-thread=N with N > 0 are set. - if(a->graph()->isOptimized() && device == DeviceType::cpu) { + if(a->graph()->isOptimized() && device == DeviceType::cpu + && a->graph()->getGemmType() == GemmType::Int16) { // dotInt16 computes A * B.T, hence the transpose for B to get A * B // if transA = false and transB = false. |