Skip to content

Commit 8d4413c

Browse files
committed
Fix comments mentioning MKL
1 parent 4948447 commit 8d4413c

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

src/graph/auto_tuner.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ class AutoTuner : public AutoTunerRecorder {
2626
const size_t collectStatMax = 50;
2727
UPtr<timer::CPUTimer> timer_;
2828

29-
// This structure holds a hash key an algorithm function (e.g. int16, packed gemm, mkl gemm)
29+
// This structure holds a hash key an algorithm function (e.g. int16, packed gemm, fp32 gemm)
3030
// for a specific operation size
3131
// hash: a unique hash key for each operation size
3232
// (e.g. m, n, k, transpose A, transpose B, bias size for GEMM)

src/graph/expression_operators.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -576,7 +576,7 @@ Expr bdot_legacy(Expr a, Expr b, bool transA, bool transB, float scale) {
576576
}
577577

578578
Expr affineDefault(Expr a, Expr b, Expr bias, bool transA, bool transB, float scale) {
579-
// general version, MKL, CBlas or CUDA
579+
// general version (see affine for packed-type variants)
580580

581581
int rows = a->shape().elements() / a->shape()[-1];
582582
Expr ones = a->graph()->ones({ rows, 1 });

src/tensors/backend.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ namespace marian {
88
// GEMM type enum
99
typedef enum {
1010
Auto = 0, // auto tuning between available GEMMs
11-
Float32 = 1, // MKL based GEMM, fp32
11+
Float32 = 1, // fp32 based GEMM
1212
FbFp16Packed = 10, // FBGEMM based fp16 GEMM with packing
1313
FbInt8Packed = 11 // FBGEMM based int8 GEMM with packing
1414
} GemmType;

0 commit comments

Comments
 (0)