comment updates

iree-org · Nov 11, 2024 · f623b7f · f623b7f
1 parent a8050a8
commit f623b7f
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 5 deletions.
diff --git a/compiler/src/iree/compiler/Codegen/LLVMCPU/KernelDispatch.cpp b/compiler/src/iree/compiler/Codegen/LLVMCPU/KernelDispatch.cpp
@@ -2918,9 +2918,13 @@ setLoweringConfigForComputeOps(mlir::FunctionOpInterface entryPointFn,
         cast<ShapedType>(rootOperation->getResultTypes().front())
             .getElementType()
             .getIntOrFloatBitWidth();
-    auto innermostTileSize = commonVecTileSizes.back();
-    commonVecTileSizes.back() =
-        llvm::alignTo(innermostTileSize * elementTypeSize, 8) / elementTypeSize;
+    // for now just enable for i1
+    if (elementTypeSize == 1) {
+      auto innermostTileSize = commonVecTileSizes.back();
+      commonVecTileSizes.back() =
+          llvm::alignTo(innermostTileSize * elementTypeSize, 8) /
+          elementTypeSize;
+    }
   }
 
   // Set the lowering configs with new tile sizes.

diff --git a/compiler/src/iree/compiler/Dialect/Stream/Transforms/test/encode_host_tensors_packing.mlir b/compiler/src/iree/compiler/Dialect/Stream/Transforms/test/encode_host_tensors_packing.mlir
@@ -1,4 +1,4 @@
-// RUN: iree-opt --split-input-file --iree-stream-encode-host-tensors --iree-experimental-packed-i1-storage --verify-diagnostics %s | FileCheck %s
+// RUN: iree-opt --split-input-file --iree-stream-encode-host-tensors --verify-diagnostics %s | FileCheck %s
 
 // CHECK-LABEL:  util.func public @denseTensorConstantI2()
 util.func public @denseTensorConstantI2() -> !stream.resource<constant> {

diff --git a/tests/e2e/subbyte_types/subbyte_types.mlir b/tests/e2e/subbyte_types/subbyte_types.mlir
@@ -11,7 +11,7 @@ func.func @i1_type() {
         ins(%lhs, %rhs : tensor<8xi1>, tensor<8xi1>) outs(%empty: tensor<8xi1>) {
   ^bb0(%inlhs: i1, %inrhs: i1, %out: i1):
     %inres = arith.xori %inlhs, %inrhs: i1
-    linalg.yield %inres : i1 
+    linalg.yield %inres : i1
   } -> tensor<8xi1>
   %tensor_res = flow.tensor.bitcast %res : tensor<8xi1> -> tensor<1xi8>
   check.expect_eq_const(%tensor_res, dense<[255]> : tensor<1xi8>) : tensor<1xi8>