add empty cpu optimization driver. refactor intrinsic selection module.

src/pystencils/backend/kernelcreation/__init__.py

@@ -10,6 +10,8 @@ from .iteration_space import (
     create_sparse_iteration_space,
 )
 
+from .cpu_optimization import optimize_cpu
+
 __all__ = [
     "KernelCreationContext",
     "KernelAnalysis",
@@ -19,4 +21,5 @@ __all__ = [
     "SparseIterationSpace",
     "create_full_iteration_space",
     "create_sparse_iteration_space",
+    "optimize_cpu",
 ]

src/pystencils/backend/kernelcreation/cpu_optimization.py

+from __future__ import annotations
+
+from .context import KernelCreationContext
+from ..platforms import GenericCpu
+from ..ast.structural import PsBlock
+
+from ...config import CpuOptimConfig
+
+
+def optimize_cpu(
+    ctx: KernelCreationContext,
+    platform: GenericCpu,
+    kernel_ast: PsBlock,
+    cfg: CpuOptimConfig,
+):
+    """Carry out CPU-specific optimizations according to the given configuration."""
+
+    if cfg.loop_blocking:
+        raise NotImplementedError("Loop blocking not implemented yet.")
+
+    if cfg.vectorize is not False:
+        raise NotImplementedError("Vectorization not implemented yet")
+
+    if cfg.openmp:
+        raise NotImplementedError("OpenMP not implemented yet")
+
+    if cfg.use_cacheline_zeroing:
+        raise NotImplementedError("CL-zeroing not implemented yet")

src/pystencils/backend/kernelcreation/freeze.py

@@ -48,22 +48,6 @@ class FreezeExpressions:
 
     TODO: Document the full set of supported SymPy features, with restrictions and caveats
     TODO: Properly document the SymPy extensions provided by pystencils
-
-    TODO: This is a (possibly incomplete) list of SymPy language features that still need to be implemented:
-
-     - Augmented Assignments
-     - AddressOf
-     - Relations (sp.Relational)
-     - pystencils.sympyextensions.integer_functions
-     - pystencils.sympyextensions.bit_masks
-     - GPU fast approximations (pystencils.fast_approximation)
-     - ConditionalFieldAccess
-     - sp.Piecewise
-     - sp.floor, sp.ceiling
-     - sp.log, sp.atan2, sp.sinh, sp.cosh. sp.atan
-     - sp.Min, sp.Max: multi-argument versions
-     - Modulus (sp.Mod)
-
     """
 
     def __init__(self, ctx: KernelCreationContext):

src/pystencils/backend/platforms/generic_cpu.py

@@ -19,7 +19,7 @@ from ..ast.expressions import (
     PsLookup,
 )
 from ...types import PsVectorType, PsCustomType
-from ..transformations.vector_intrinsics import IntrinsicOps
+from ..transformations.select_intrinsics import IntrinsicOps
 
 
 class GenericCpu(Platform):

src/pystencils/backend/platforms/x86.py

@@ -9,7 +9,7 @@ from ..ast.expressions import (
     PsAddressOf,
     PsSubscript,
 )
-from ..transformations.vector_intrinsics import IntrinsicOps
+from ..transformations.select_intrinsics import IntrinsicOps
 from ...types import PsCustomType, PsVectorType
 from ..constants import PsConstant
 

src/pystencils/backend/transformations/__init__.py

 from .eliminate_constants import EliminateConstants
 from .erase_anonymous_structs import EraseAnonymousStructTypes
-from .vector_intrinsics import MaterializeVectorIntrinsics
+from .select_intrinsics import MaterializeVectorIntrinsics
 
 __all__ = [
     "EliminateConstants",

src/pystencils/kernelcreation.py

@@ -99,10 +99,10 @@ def create_kernel(
     erase_anons = EraseAnonymousStructTypes(ctx)
     kernel_ast = cast(PsBlock, erase_anons(kernel_ast))
 
-    #   7. Apply optimizations
-    #     - Vectorization
-    #     - OpenMP
-    #     - Loop Splitting, Tiling, Blocking
+    #   Target-Specific optimizations
+    if config.target.is_cpu() and config.cpu_optim is not None:
+        from .backend.kernelcreation import optimize_cpu
+        optimize_cpu(ctx, platform, kernel_ast, config.cpu_optim)
 
     assert config.jit is not None
     return create_kernel_function(