try unrolling tiled bands

source/codegen/auto-scheduler/ast-parser.lisp

@@ -73,6 +73,7 @@ scop.lisp for the opposite things.
           ;; UNROLL_BODY is triggered by the UNROLL_PARENT. Without it the form is ignored.
           (assert (null (astfor-marks user)) () "UNROLL_BODY should be orthogonal with other directives.")
           (setf (astfor-marks user) (list mark)))
+         ((equalp mark "TILE_BAND"))
          (T
           (warn "mark: ignored the mark ~a for ~a" mark user))))
       (otherwise
@@ -207,7 +208,7 @@ scop.lisp for the opposite things.
 		 ((ASTBlock :body body) (map 'list #'lower body))
 		 ((AstFor :idx idx :from upfrom :to to :by by :body body :scope scope)
                   ;; remove an empty loop
-                  (when (not (expr-scalar-equivalent-p upfrom (expr-detach-loop-bound to)))
+g                  (when (not (expr-scalar-equivalent-p upfrom (expr-detach-loop-bound to)))
                     (push idx space)
 		    (push (r/for idx upfrom to by scope) new-graph)
 		    (lower body)

source/codegen/auto-scheduler/auto-scheduler.lisp

@@ -69,10 +69,11 @@ So we are going to apply an optiimzation method which takes a long time to searc
   (print "REUSE")
   ;; Data Reuse kernel has a chance to apply the tiling
   ;; And Tuning the tiling size tested by the measurer
-  ;; (caten/codegen/tiling::apply-tile (getattr node :polyhedral) 16)
+  (caten/codegen/tiling::apply-tile (getattr node :polyhedral) 64) ;; Tile_Size = Optimal_Tile_Size * N_UNROLL
   (caten/codegen/unroll:apply-unroll node 4)
   ;; (caten/codegen/unroll::apply-unroll node 4)
   ;; [TODO] After Tiling, Unroll the outermost tile band.
+  ;; [TODO] If the outermost loop is a tile -> #pragma omp parallel for collapse(2)
   (caten/codegen/coincidence:apply-parallel
    (getattr node :polyhedral)
    (auto-scheduler-n-global-loops auto-scheduler)))

source/codegen/auto-scheduler/tiling.lisp

@@ -60,8 +60,10 @@ dims is used to specify the tiling sizes for each dimension."
       (shift-band-zero band)
     (let* ((tiling-sizes (tiling-sizes band :size-default size-default :dims dims))
            (partial-schedule (tile-partial-schedule partial-schedule tiling-sizes))
-           (tiled-sched (multi-union-pw-aff-add partial-schedule shift)))
-      (schedule-node-insert-partial-schedule band tiled-sched))))
+           (tiled-sched (multi-union-pw-aff-add partial-schedule shift))
+           (tiled-sched (schedule-node-insert-partial-schedule band tiled-sched))
+           (tiled-sched (schedule-node-insert-mark tiled-sched (isl::make-id-from-str "TILE_BAND"))))
+      tiled-sched)))
 
 (defun get-tileable-bands (poly)
   (map-schedule-nodes

source/codegen/auto-scheduler/unroll.lisp

@@ -9,7 +9,9 @@
 
 (defun mark-unroll-p (mark)
   (declare (type string mark))
-  (scan "UNROLL" (string-upcase mark)))
+  (or
+   (scan "TILE_BAND" (string-upcase mark))
+   (scan "UNROLL" (string-upcase mark))))
 
 (defun mark-unroll-body-p (mark) (scan "UNROLL_BODY" (string-upcase mark)))
 (defun mark-unroll-parent-p (mark) (scan "UNROLL_PARENT" (string-upcase mark)))