Skip to content

Commit

Permalink
Register all tensor tiling ops
Browse files Browse the repository at this point in the history
Previously, we would single out tensor.pad which would incorrectly make
all transforms in IREE think that this is not an op that supports TilingInterface,
in violation of the upstream semantics.
  • Loading branch information
nicolasvasilache committed Apr 12, 2023
1 parent 9466b5e commit 2d0fa54
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion compiler/src/iree/compiler/Tools/init_mlir_dialects.h
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,7 @@ inline void registerMlirDialects(DialectRegistry &registry) {
shape::ShapeDialect>();
// clang-format on
tensor::registerInferTypeOpInterfaceExternalModels(registry);
tensor::registerTilingInterfaceExternalModelsForPackUnPackOps(registry);
tensor::registerTilingInterfaceExternalModels(registry);

#ifdef IREE_HAVE_C_OUTPUT_FORMAT
registry.insert<emitc::EmitCDialect>();
Expand Down

0 comments on commit 2d0fa54

Please sign in to comment.