This is awesome! Thank you very much. I'll wait for the device then. 😎

The problem is with broadcasted template. Tensor with "static" shape works OK.

It seems, that this is indeed impossible due to MLIR dialect for Structured Control Flow. https://llvm.discourse.group/t/scf-whileop-type-inference-with-dynamic-shapes/2529

extension MPSDataType {     func description() -> String {         switch(self) {         case .float32:             return "f32"         case .int32:             return "i32"         // other...         default:             return "?"         }     } }