From 7009ea562f142a98a63fd086a03c9bbd32b14b3b Mon Sep 17 00:00:00 2001
From: Michael Lange <Michael.Lange@ecmwf.int>
Date: Wed, 8 May 2024 15:04:27 +0000
Subject: [PATCH] Loki-transform: Allow file configuration for SCC
 transformations

We're leaving the defaults in for the moment until the full hit comes.
---
 scripts/loki_transform.py | 55 ++++++++++++++++++++++-----------------
 1 file changed, 31 insertions(+), 24 deletions(-)

diff --git a/scripts/loki_transform.py b/scripts/loki_transform.py
index c48274447..f26101cf4 100644
--- a/scripts/loki_transform.py
+++ b/scripts/loki_transform.py
@@ -280,39 +280,46 @@ def transform_subroutine(self, routine, **kwargs):
         scheduler.process( pipeline )
 
     if mode == 'scc':
-        pipeline = SCCVectorPipeline(
-            horizontal=horizontal,
-            block_dim=block_dim, directive=directive,
-            trim_vector_sections=trim_vector_sections
-        )
+        pipeline = scheduler.config.transformations.get('scc', None)
+        if not pipeline:
+            pipeline = SCCVectorPipeline(
+                horizontal=horizontal,
+                block_dim=block_dim, directive=directive,
+                trim_vector_sections=trim_vector_sections
+            )
         scheduler.process( pipeline )
 
     if mode == 'scc-hoist':
-        pipeline = SCCHoistPipeline(
-            horizontal=horizontal,
-            block_dim=block_dim, directive=directive,
-            dim_vars=(vertical.size,) if vertical else None,
-            trim_vector_sections=trim_vector_sections
-        )
+        pipeline = scheduler.config.transformations.get('scc-hoist', None)
+        if not pipeline:
+            pipeline = SCCHoistPipeline(
+                horizontal=horizontal,
+                block_dim=block_dim, directive=directive,
+                dim_vars=(vertical.size,) if vertical else None,
+                trim_vector_sections=trim_vector_sections
+            )
         scheduler.process( pipeline )
 
     if mode == 'scc-stack':
-        pipeline = SCCStackPipeline(
-            horizontal=horizontal,
-            block_dim=block_dim, directive=directive,
-            check_bounds=False,
-            trim_vector_sections=trim_vector_sections
-        )
+        pipeline = scheduler.config.transformations.get('scc-stack', None)
+        if not pipeline:
+            pipeline = SCCStackPipeline(
+                horizontal=horizontal,
+                block_dim=block_dim, directive=directive,
+                check_bounds=False,
+                trim_vector_sections=trim_vector_sections
+            )
         scheduler.process( pipeline )
 
     if mode == 'scc-raw-stack':
-        pipeline = SCCStackPipeline(
-            horizontal=horizontal,
-            block_dim=block_dim, directive=directive,
-            check_bounds=False,
-            trim_vector_sections=trim_vector_sections,
-            driver_horizontal='NPROMA'
-        )
+        pipeline = scheduler.config.transformations.get('scc-raw-stack', None)
+        if not pipeline:
+            pipeline = SCCStackPipeline(
+                horizontal=horizontal,
+                block_dim=block_dim, directive=directive,
+                check_bounds=False,
+                trim_vector_sections=trim_vector_sections,
+            )
         scheduler.process( pipeline )
 
     if mode in ['cuf-parametrise', 'cuf-hoist', 'cuf-dynamic']: