Resnet is fast

2026-07-20 11:34:59 +02:00
parent 5f42da36ae
commit 6bad9a8008
16 changed files with 883 additions and 138 deletions
@@ -270,6 +270,25 @@ def SpatReluPlanOp : SpatOp<"relu_plan", []> {
  let hasVerifier = 1;
 }

+def SpatMaxPool2DPlanOp : SpatOp<"max_pool2d_plan", []> {
+  let summary = "Layout-aware 2D NCHW MaxPool planning op";
+
+  let arguments = (ins
+    SpatTensor:$input,
+    DenseI64ArrayAttr:$kernelShape,
+    DenseI64ArrayAttr:$pads,
+    DenseI64ArrayAttr:$strides,
+    DenseI64ArrayAttr:$dilations,
+    StrAttr:$logicalLayout
+  );
+
+  let results = (outs
+    SpatTensor:$output
+  );
+
+  let hasVerifier = 1;
+}
+
 def SpatBiasAddPlanOp : SpatOp<"bias_add_plan", []> {
  let summary = "Layout-aware Conv-style bias add planning op";