fix gemm tutorial for env that may not have right instruction (#810)

adceea22 · Tianqi Chen · GitHub · 45138e9d · adceea22
Commit adceea22 authored Jan 23, 2018 by Tianqi Chen Committed by GitHub Jan 23, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletions

tutorials/optimize/opt_gemm.py
+3 -1

No files found.
--- a/tutorials/optimize/opt_gemm.py
+++ b/tutorials/optimize/opt_gemm.py
@@ -52,7 +52,9 @@ N = 1024
 dtype = "float32"

 # using Intel AVX2(Advanced Vector Extensions) ISA for SIMD
-target = 'llvm -mcpu=core-avx2'
+# To get the best performance, please change the following line
+# to llvm -mcpu=core-avx2, or specific type of CPU you use
+target = 'llvm'
 ctx = tvm.context(target, 0)

 # Random generated tensor for testing