Refactor compiler -> snaxc

compiler/accelerators/accelerator.py → snaxc/compiler/accelerators/accelerator.py

@@ -3,7 +3,7 @@
 
 from xdsl.ir import Operation
 
-from compiler.dialects import accfg
+from snaxc.dialects import accfg
 
 
 class Accelerator(ABC):

compiler/accelerators/dispatching.py → snaxc/compiler/accelerators/dispatching.py

@@ -4,8 +4,8 @@
 
 from xdsl.ir import Attribute
 
-from compiler.accelerators.accelerator import Accelerator
-from compiler.dialects.kernel import KernelOp
+from snaxc.accelerators.accelerator import Accelerator
+from snaxc.dialects.kernel import KernelOp
 
 
 @dataclass

compiler/accelerators/gemmini.py → snaxc/compiler/accelerators/gemmini.py

@@ -4,9 +4,9 @@
 from xdsl.dialects.builtin import IndexType, i64
 from xdsl.ir import Operation
 
-from compiler.accelerators.rocc import RoCCAccelerator
-from compiler.dialects import accfg
-from compiler.util.pack_bitlist import pack_bitlist
+from snaxc.accelerators.rocc import RoCCAccelerator
+from snaxc.dialects import accfg
+from snaxc.util.pack_bitlist import pack_bitlist
 
 
 class GemminiAccelerator(RoCCAccelerator):

compiler/accelerators/registry.py → snaxc/compiler/accelerators/registry.py

@@ -2,14 +2,14 @@
 
 from xdsl.dialects.builtin import ModuleOp, StringAttr
 
-from compiler.accelerators import find_accelerator_op
-from compiler.accelerators.accelerator import Accelerator
-from compiler.accelerators.gemmini import GemminiAccelerator
-from compiler.accelerators.snax_alu import SNAXAluAccelerator
-from compiler.accelerators.snax_gemm import SNAXGEMMAccelerator
-from compiler.accelerators.snax_gemmx import SNAXGEMMXAccelerator
-from compiler.accelerators.snax_hwpe_mult import SNAXHWPEMultAccelerator
-from compiler.dialects.accfg import AcceleratorOp
+from snaxc.accelerators import find_accelerator_op
+from snaxc.accelerators.accelerator import Accelerator
+from snaxc.accelerators.gemmini import GemminiAccelerator
+from snaxc.accelerators.snax_alu import SNAXAluAccelerator
+from snaxc.accelerators.snax_gemm import SNAXGEMMAccelerator
+from snaxc.accelerators.snax_gemmx import SNAXGEMMXAccelerator
+from snaxc.accelerators.snax_hwpe_mult import SNAXHWPEMultAccelerator
+from snaxc.dialects.accfg import AcceleratorOp
 
 
 class AcceleratorRegistry:

compiler/accelerators/rocc.py → snaxc/compiler/accelerators/rocc.py

@@ -5,9 +5,9 @@
 from xdsl.dialects.builtin import IntegerAttr, i64
 from xdsl.ir import Operation, SSAValue
 
-from compiler.accelerators.accelerator import Accelerator
-from compiler.dialects import accfg
-from compiler.inference.trace_acc_state import infer_state_of
+from snaxc.accelerators.accelerator import Accelerator
+from snaxc.dialects import accfg
+from snaxc.inference.trace_acc_state import infer_state_of
 
 
 class RoCCAccelerator(Accelerator, ABC):

compiler/accelerators/snax.py → snaxc/compiler/accelerators/snax.py

@@ -7,13 +7,13 @@
 from xdsl.dialects.scf import ConditionOp, WhileOp, YieldOp
 from xdsl.ir import Operation, OpResult, SSAValue
 
-from compiler.accelerators.accelerator import Accelerator
-from compiler.accelerators.streamers import StreamerConfiguration
-from compiler.accelerators.streamers.streamers import StreamerFlag, StreamerOpts
-from compiler.dialects import accfg
-from compiler.dialects.dart import StreamingRegionOpBase
-from compiler.dialects.snax_stream import StreamerConfigurationAttr, StreamingRegionOp
-from compiler.ir.dart.access_pattern import Template
+from snaxc.accelerators.accelerator import Accelerator
+from snaxc.accelerators.streamers import StreamerConfiguration
+from snaxc.accelerators.streamers.streamers import StreamerFlag, StreamerOpts
+from snaxc.dialects import accfg
+from snaxc.dialects.dart import StreamingRegionOpBase
+from snaxc.dialects.snax_stream import StreamerConfigurationAttr, StreamingRegionOp
+from snaxc.ir.dart.access_pattern import Template
 
 c0_attr = builtin.IntegerAttr(0, builtin.IndexType())
 

compiler/accelerators/snax_alu.py → snaxc/compiler/accelerators/snax_alu.py

@@ -5,20 +5,20 @@
 from xdsl.ir import Operation, SSAValue
 from xdsl.ir.affine import AffineMap
 
-import compiler.dialects.kernel as kernel
-from compiler.accelerators.dispatching import DispatchTemplate, SupportedKernel
-from compiler.accelerators.snax import (
+import snaxc.dialects.kernel as kernel
+from snaxc.accelerators.dispatching import DispatchTemplate, SupportedKernel
+from snaxc.accelerators.snax import (
     SNAXAccelerator,
     SNAXPollingBarrier3,
     SNAXStreamer,
 )
-from compiler.accelerators.streamers import (
+from snaxc.accelerators.streamers import (
     Streamer,
     StreamerConfiguration,
     StreamerType,
 )
-from compiler.dialects import accfg, dart, snax_stream
-from compiler.ir.dart.access_pattern import Template, TemplatePattern
+from snaxc.dialects import accfg, dart, snax_stream
+from snaxc.ir.dart.access_pattern import Template, TemplatePattern
 
 default_streamer = StreamerConfiguration(
     [

compiler/accelerators/snax_gemm.py → snaxc/compiler/accelerators/snax_gemm.py

@@ -5,16 +5,16 @@
 from xdsl.ir import Operation, SSAValue
 from xdsl.ir.affine import AffineDimExpr, AffineMap
 
-import compiler.dialects.kernel as kernel
-from compiler.accelerators.dispatching import DispatchTemplate, SupportedKernel
-from compiler.accelerators.snax import SNAXAccelerator, SNAXStreamer
-from compiler.accelerators.streamers import (
+import snaxc.dialects.kernel as kernel
+from snaxc.accelerators.dispatching import DispatchTemplate, SupportedKernel
+from snaxc.accelerators.snax import SNAXAccelerator, SNAXStreamer
+from snaxc.accelerators.streamers import (
     Streamer,
     StreamerConfiguration,
     StreamerType,
 )
-from compiler.dialects import accfg, dart, snax_stream
-from compiler.ir.dart.access_pattern import Template, TemplatePattern
+from snaxc.dialects import accfg, dart, snax_stream
+from snaxc.ir.dart.access_pattern import Template, TemplatePattern
 
 default_streamer = StreamerConfiguration(
     [

compiler/accelerators/snax_gemmx.py → snaxc/compiler/accelerators/snax_gemmx.py

@@ -6,21 +6,21 @@
 from xdsl.ir import BlockArgument, Operation, SSAValue
 from xdsl.ir.affine import AffineDimExpr, AffineMap
 
-from compiler.accelerators.dispatching import DispatchTemplate, SupportedKernel
-from compiler.accelerators.snax import (
+from snaxc.accelerators.dispatching import DispatchTemplate, SupportedKernel
+from snaxc.accelerators.snax import (
     SNAXAccelerator,
     SNAXPollingBarrier4,
     SNAXStreamer,
 )
-from compiler.accelerators.streamers import (
+from snaxc.accelerators.streamers import (
     Streamer,
     StreamerConfiguration,
     StreamerType,
 )
-from compiler.accelerators.streamers.streamers import StreamerOpts
-from compiler.dialects import accfg, dart, kernel, snax_stream
-from compiler.ir.dart.access_pattern import Template, TemplatePattern
-from compiler.util.pack_bitlist import pack_bitlist
+from snaxc.accelerators.streamers.streamers import StreamerOpts
+from snaxc.dialects import accfg, dart, kernel, snax_stream
+from snaxc.ir.dart.access_pattern import Template, TemplatePattern
+from snaxc.util.pack_bitlist import pack_bitlist
 
 default_streamer = StreamerConfiguration(
     [

compiler/accelerators/snax_hwpe_mult.py → snaxc/compiler/accelerators/snax_hwpe_mult.py

@@ -4,8 +4,8 @@
 from xdsl.ir import Attribute, Operation, SSAValue
 from xdsl.utils.hints import isa
 
-from compiler.accelerators.snax import SNAXAccelerator, SNAXPollingBarrier
-from compiler.dialects import accfg
+from snaxc.accelerators.snax import SNAXAccelerator, SNAXPollingBarrier
+from snaxc.dialects import accfg
 
 
 class SNAXHWPEMultAccelerator(SNAXAccelerator, SNAXPollingBarrier):

compiler/accelerators/util.py → snaxc/compiler/accelerators/util.py

@@ -2,7 +2,7 @@
 from xdsl.ir import Operation
 from xdsl.traits import SymbolTable
 
-from compiler.dialects.accfg import AcceleratorOp
+from snaxc.dialects.accfg import AcceleratorOp
 
 
 def find_accelerator_op(

compiler/dialects/__init__.py → snaxc/compiler/dialects/__init__.py

@@ -7,37 +7,37 @@ def get_all_snax_dialects() -> dict[str, Callable[[], Dialect]]:
     """Returns all available snax dialects"""
 
     def get_accfg():
-        from compiler.dialects.accfg import ACCFG
+        from snaxc.dialects.accfg import ACCFG
 
         return ACCFG
 
     def get_dart():
-        from compiler.dialects.dart import Dart
+        from snaxc.dialects.dart import Dart
 
         return Dart
 
     def get_debug():
-        from compiler.dialects.test.debug import Debug
+        from snaxc.dialects.test.debug import Debug
 
         return Debug
 
     def get_kernel():
-        from compiler.dialects.kernel import Kernel
+        from snaxc.dialects.kernel import Kernel
 
         return Kernel
 
     def get_snax():
-        from compiler.dialects.snax import Snax
+        from snaxc.dialects.snax import Snax
 
         return Snax
 
     def get_snax_stream():
-        from compiler.dialects.snax_stream import SnaxStream
+        from snaxc.dialects.snax_stream import SnaxStream
 
         return SnaxStream
 
     def get_tsl():
-        from compiler.dialects.tsl import TSL
+        from snaxc.dialects.tsl import TSL
 
         return TSL
 

compiler/dialects/snax.py → snaxc/compiler/dialects/snax.py

@@ -31,14 +31,14 @@
 from xdsl.printer import Printer
 from xdsl.utils.exceptions import VerifyException
 
-from compiler.accelerators.streamers import (
+from snaxc.accelerators.streamers import (
     Streamer,
     StreamerConfiguration,
     StreamerFlag,
     StreamerType,
 )
-from compiler.accelerators.streamers.streamers import StreamerOpts
-from compiler.util.memref_descriptor import LLVMMemrefDescriptor
+from snaxc.accelerators.streamers.streamers import StreamerOpts
+from snaxc.util.memref_descriptor import LLVMMemrefDescriptor
 
 
 @irdl_op_definition

compiler/dialects/snax_stream.py → snaxc/compiler/dialects/snax_stream.py

@@ -25,9 +25,9 @@
 from xdsl.parser import AttrParser
 from xdsl.printer import Printer
 
-from compiler.accelerators import find_accelerator_op
-from compiler.accelerators.streamers import StreamerConfiguration
-from compiler.dialects.snax import StreamerConfigurationAttr
+from snaxc.accelerators import find_accelerator_op
+from snaxc.accelerators.streamers import StreamerConfiguration
+from snaxc.dialects.snax import StreamerConfigurationAttr
 
 
 @irdl_attr_definition

compiler/dialects/tsl.py → snaxc/compiler/dialects/tsl.py

@@ -20,8 +20,8 @@
 from xdsl.parser import AttrParser
 from xdsl.printer import Printer
 
-from compiler.ir.tsl import TiledStridedLayout
-from compiler.parser.tsl_parser import TSLParser
+from snaxc.ir.tsl import TiledStridedLayout
+from snaxc.parser.tsl_parser import TSLParser
 
 
 @irdl_attr_definition

compiler/inference/helpers.py → snaxc/compiler/inference/helpers.py

@@ -3,7 +3,7 @@
 from xdsl.dialects import func, llvm, scf
 from xdsl.ir import Block, BlockArgument, Operation, OpResult, Region, SSAValue
 
-from compiler.dialects import accfg
+from snaxc.dialects import accfg
 
 
 def has_accfg_effects(op: Operation) -> bool:

compiler/inference/scoped_setups.py → snaxc/compiler/inference/scoped_setups.py

@@ -56,8 +56,8 @@
 from xdsl.rewriter import InsertPoint
 from xdsl.traits import is_side_effect_free
 
-from compiler.dialects import accfg
-from compiler.inference.helpers import val_is_defined_in_block
+from snaxc.dialects import accfg
+from snaxc.inference.helpers import val_is_defined_in_block
 
 
 def get_scoped_setup_inputs(

compiler/inference/trace_acc_state.py → snaxc/compiler/inference/trace_acc_state.py

@@ -18,7 +18,7 @@
 from xdsl.dialects import scf
 from xdsl.ir import Block, BlockArgument, Region, SSAValue
 
-from compiler.dialects import accfg
+from snaxc.dialects import accfg
 
 State = dict[str, SSAValue]
 

compiler/ir/dart/access_pattern.py → snaxc/compiler/ir/dart/access_pattern.py

@@ -6,7 +6,7 @@
 from typing_extensions import Self, TypeVar, overload
 from xdsl.ir.affine import AffineDimExpr, AffineMap
 
-from compiler.ir.dart.affine_transform import AffineTransform
+from snaxc.ir.dart.affine_transform import AffineTransform
 
 
 @dataclass(frozen=True)

compiler/ir/dart/scheduler.py → snaxc/compiler/ir/dart/scheduler.py

@@ -2,7 +2,7 @@
 
 import numpy as np
 
-from compiler.ir.dart.access_pattern import Schedule, Template
+from snaxc.ir.dart.access_pattern import Schedule, Template
 
 
 def scheduler_backtrack(

snaxc/compiler/ir/tsl/__init__.py

@@ -0,0 +1,5 @@
+from snaxc.ir.tsl.stride import Stride
+from snaxc.ir.tsl.tiled_stride import TiledStride
+from snaxc.ir.tsl.tiled_strided_layout import TiledStridedLayout
+
+__all__ = ["Stride", "TiledStride", "TiledStridedLayout"]

compiler/ir/tsl/tiled_stride.py → snaxc/compiler/ir/tsl/tiled_stride.py

@@ -4,7 +4,7 @@
 from dataclasses import dataclass
 from typing import Self
 
-from compiler.ir.tsl.stride import Stride
+from snaxc.ir.tsl.stride import Stride
 
 
 @dataclass

compiler/ir/tsl/tiled_strided_layout.py → snaxc/compiler/ir/tsl/tiled_strided_layout.py

@@ -7,8 +7,8 @@
 import numpy as np
 from numpy._typing import NDArray
 
-from compiler.ir.tsl.stride import Stride
-from compiler.ir.tsl.tiled_stride import TiledStride
+from snaxc.ir.tsl.stride import Stride
+from snaxc.ir.tsl.tiled_stride import TiledStride
 
 
 @dataclass(frozen=True)

compiler/parser/tsl_parser.py → snaxc/compiler/parser/tsl_parser.py

@@ -5,7 +5,7 @@
 from xdsl.utils.exceptions import ParseError
 from xdsl.utils.mlir_lexer import MLIRTokenKind
 
-from compiler.ir.tsl import Stride, TiledStride, TiledStridedLayout
+from snaxc.ir.tsl import Stride, TiledStride, TiledStridedLayout
 
 
 class TSLParser(BaseParser):

compiler/tools/snax_opt_main.py → snaxc/compiler/tools/snax_opt_main.py

@@ -6,8 +6,8 @@
 from xdsl.transforms import get_all_passes
 from xdsl.xdsl_opt_main import xDSLOptMain
 
-from compiler.dialects import get_all_snax_dialects
-from compiler.transforms import get_all_snax_passes
+from snaxc.dialects import get_all_snax_dialects
+from snaxc.transforms import get_all_snax_passes
 
 
 class SNAXOptMain(xDSLOptMain):