From 9cd93baa73efb31ad19edb0204e6a4b43847c6d9 Mon Sep 17 00:00:00 2001
From: Lei Zhang <antiagainst@google.com>
Date: Thu, 1 Apr 2021 11:48:38 -0400
Subject: [PATCH] Turn vectorization on by default for linalg on tensors path
 (#5280)

---
 iree/compiler/Conversion/LinalgToSPIRV/CodeGenOptionUtils.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/iree/compiler/Conversion/LinalgToSPIRV/CodeGenOptionUtils.cpp b/iree/compiler/Conversion/LinalgToSPIRV/CodeGenOptionUtils.cpp
index b633cfe22dab..7036ae190f0b 100644
--- a/iree/compiler/Conversion/LinalgToSPIRV/CodeGenOptionUtils.cpp
+++ b/iree/compiler/Conversion/LinalgToSPIRV/CodeGenOptionUtils.cpp
@@ -56,7 +56,8 @@ SPIRVCodegenOptions getSPIRVCodegenOptionsFromClOptions() {
   options.workgroupSize.assign(clWorkgroupSizes.begin(),
                                clWorkgroupSizes.end());
   options.tileSizes.assign(clTileSizes.begin(), clTileSizes.end());
-  options.enableVectorization = clEnableVectorization;
+  options.enableVectorization =
+      clEnableLinalgOnTensorsSPIRV || clEnableVectorization;
   options.useWorkgroupMemory = clUseWorkgroupMemory;
   options.vectorizeMemref = clVectorizeMemref;
   options.usingLinalgOnTensors = clEnableLinalgOnTensorsSPIRV;