Skip to main content

oxilean_codegen/mlir_backend/
affinemap_traits.rs

1//! # AffineMap - Trait Implementations
2//!
3//! This module contains trait implementations for `AffineMap`.
4//!
5//! ## Implemented Traits
6//!
7//! - `Display`
8//!
9//! 🤖 Generated with [SplitRS](https://github.com/cool-japan/splitrs)
10
11use super::types::AffineMap;
12use std::fmt;
13
14impl fmt::Display for AffineMap {
15    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
16        match self {
17            AffineMap::Identity(n) => {
18                let dims: Vec<String> = (0..*n).map(|i| format!("d{}", i)).collect();
19                let dims_str = dims.join(", ");
20                write!(f, "affine_map<({}) -> ({})>", dims_str, dims_str)
21            }
22            AffineMap::Constant => write!(f, "affine_map<() -> ()>"),
23            AffineMap::Custom(s) => write!(f, "affine_map<{}>", s),
24        }
25    }
26}