mlir: fix regions view

lifting-bits · Aug 15, 2024 · 53b1a07 · 53b1a07
1 parent 107767b
commit 53b1a07
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 14 deletions.
diff --git a/mlir/include/gap/mlir/views.hpp b/mlir/include/gap/mlir/views.hpp
@@ -23,9 +23,18 @@
 
 namespace gap::mlir::views
 {
-    auto regions(operation op) -> decltype(std::views::all(op->getRegions()));
-    auto blocks(operation op) -> decltype(std::views::join(regions(op)));
-    auto operations(operation op) -> decltype(std::views::join(blocks(op)));
+    static inline auto regions(operation op) -> decltype(auto) {
+        auto regs = op->getRegions();
+        return std::ranges::subrange{regs.begin(), regs.end()};
+    }
+
+    static inline auto blocks(operation op) -> decltype(auto) {
+        return regions(op) | std::views::join;
+    }
+
+    static inline auto operations(operation op) -> decltype(auto) {
+        return blocks(op) | std::views::join;
+    }
 
     template< typename T >
     auto isa = std::views::filter(::gap::mlir::isa< T >);

diff --git a/mlir/src/views.cpp b/mlir/src/views.cpp
@@ -2,16 +2,5 @@
 
 namespace gap::mlir::views
 {
-    auto regions(operation op) -> decltype(std::views::all(op->getRegions())) {
-        return op->getRegions() | std::views::all;
-    }
-
-    auto blocks(operation op) -> decltype(std::views::join(regions(op))) {
-        return regions(op) | std::views::join;
-    }
-
-    auto operations(operation op) -> decltype(std::views::join(blocks(op))) {
-        return blocks(op) | std::views::join;
-    }
 
 } // namespace gap::mlir::views