Rengz isl bindings work

accelforge/_deprecate/_simanneal/evalmapping.py

@@ -1,11 +1,11 @@
 from collections import defaultdict
 import itertools
 import time
-from fastfusion._accelerated_imports import pd
-from fastfusion.mapper.FFM._join_pmappings.sim import PmappingGroup, Loop, Compatibility
-from fastfusion.mapper.FFM._join_pmappings.pmapping_group import PmappingDataframe
-from fastfusion.mapper.simanneal.mapspaceglobals import MapspaceGlobals
-from fastfusion.util._frozenset import fzs
+from accelforge._accelerated_imports import pd
+from accelforge.mapper.FFM._join_pmappings.sim import PmappingGroup, Loop, Compatibility
+from accelforge.mapper.FFM._join_pmappings.pmapping_group import PmappingDataframe
+from accelforge.mapper.simanneal.mapspaceglobals import MapspaceGlobals
+from accelforge.util._frozenset import fzs
 
 
 def mapping2sims(einsum_to_result: Compatibility):

accelforge/_deprecate/_simanneal/mapspaceglobals.py

@@ -1,12 +1,12 @@
 from collections import defaultdict
 import itertools
 
-from fastfusion.frontend import arch
-from fastfusion.frontend.spec import Spec
-from fastfusion.mapper.FFM._join_pmappings.join_pmappings import PmappingGroup
-from fastfusion.mapper.FFM._join_pmappings.compatibility import Loop, Compatibility
-from fastfusion.util._frozenset import fzs
-from fastfusion.mapper.FFM._join_pmappings.join_pmappings import (
+from accelforge.frontend import arch
+from accelforge.frontend.spec import Spec
+from accelforge.mapper.FFM._join_pmappings.join_pmappings import PmappingGroup
+from accelforge.mapper.FFM._join_pmappings.compatibility import Loop, Compatibility
+from accelforge.util._frozenset import fzs
+from accelforge.mapper.FFM._join_pmappings.join_pmappings import (
     make_full_equivalent_rank_variables,
 )
 

accelforge/_deprecate/_simanneal/simanneal.py

@@ -4,20 +4,20 @@
 import random
 import threading
 import time
-from fastfusion._accelerated_imports import pd
-from fastfusion.mapper.simanneal.evalmapping import quick_join
-from fastfusion.mapper.simanneal.tracking import EvaluationsScoreTracker
-from fastfusion.mapper.FFM._join_pmappings.join_pmappings import PmappingGroup
-from fastfusion.mapper.FFM._join_pmappings.compatibility import (
+from accelforge._accelerated_imports import pd
+from accelforge.mapper.simanneal.evalmapping import quick_join
+from accelforge.mapper.simanneal.tracking import EvaluationsScoreTracker
+from accelforge.mapper.FFM._join_pmappings.join_pmappings import PmappingGroup
+from accelforge.mapper.FFM._join_pmappings.compatibility import (
     TensorReservation,
     Compatibility,
 )
-from fastfusion.mapper.FFM._join_pmappings.pmapping_group import (
+from accelforge.mapper.FFM._join_pmappings.pmapping_group import (
     MAPPING_COLUMN,
     PmappingDataframe,
 )
-from fastfusion.util._frozenset import fzs
-from fastfusion.mapper.simanneal.mapspaceglobals import MapspaceGlobals
+from accelforge.util._frozenset import fzs
+from accelforge.mapper.simanneal.mapspaceglobals import MapspaceGlobals
 
 OBJECTIVE_COLUMN = None  # None -> Product
 

accelforge/_deprecate/_simanneal/wrappers.py

@@ -3,18 +3,18 @@
 import itertools
 import time
 from joblib import delayed
-from fastfusion._accelerated_imports import pd
-from fastfusion.frontend import arch
-from fastfusion.frontend.spec import Spec
-from fastfusion.mapper.FFM._join_pmappings.sim import PmappingGroup, Loop, Compatibility
-from fastfusion.mapper.FFM._join_pmappings.pmapping_group import (
+from accelforge._accelerated_imports import pd
+from accelforge.frontend import arch
+from accelforge.frontend.spec import Spec
+from accelforge.mapper.FFM._join_pmappings.sim import PmappingGroup, Loop, Compatibility
+from accelforge.mapper.FFM._join_pmappings.pmapping_group import (
     PmappingDataframe,
     is_reservation_col,
 )
-from fastfusion.mapper.simanneal.simanneal import MapspaceGlobals, _fuse_sims
-from fastfusion.mapper.simanneal.tracking import EvaluationsScoreTracker
-from fastfusion.util._frozenset import fzs
-from fastfusion.util.parallel import parallel, util
+from accelforge.mapper.simanneal.simanneal import MapspaceGlobals, _fuse_sims
+from accelforge.mapper.simanneal.tracking import EvaluationsScoreTracker
+from accelforge.util._frozenset import fzs
+from accelforge.util.parallel import parallel, util
 
 
 def mapping2sims(einsum_to_result: Compatibility):

accelforge/_deprecate/_simanneal2/__init__.py

@@ -1,7 +1,7 @@
-from fastfusion.mapper.FFM.main import (
+from accelforge.mapper.FFM.main import (
     make_pmappings,
     MultiEinsumPmappings,
     Mappings,
 )
-from fastfusion.frontend.mapper.metrics import Metrics
+from accelforge.frontend.mapper.metrics import Metrics
 from .simanneal import join_pmappings

accelforge/_deprecate/_simanneal2/simanneal.py

@@ -2,29 +2,29 @@
 import os
 import random
 from typing import Callable, Generator
-from fastfusion import arch, util
-from fastfusion import Spec
-from fastfusion.frontend.mapper.metrics import Metrics
-from fastfusion.mapper.FFM.pmappings import MultiEinsumPmappings
-from fastfusion.mapper.FFM._join_pmappings.compress_pmappings import (
+from accelforge import arch, util
+from accelforge import Spec
+from accelforge.frontend.mapper.metrics import Metrics
+from accelforge.mapper.FFM.pmappings import MultiEinsumPmappings
+from accelforge.mapper.FFM._join_pmappings.compress_pmappings import (
     compress_einsum2pmappings,
     decompress_pmappings,
 )
-from fastfusion.frontend.workload import EinsumName
-from fastfusion.frontend.mapping import Mapping
-from fastfusion.mapper.FFM import PmappingGroup
-from fastfusion.mapper.FFM._pareto_df.df_convention import (
+from accelforge.frontend.workload import EinsumName
+from accelforge.frontend.mapping import Mapping
+from accelforge.mapper.FFM import PmappingGroup
+from accelforge.mapper.FFM._pareto_df.df_convention import (
     MAPPING_COLUMN,
     col2nameloop,
 )
-from fastfusion.mapper.FFM._join_pmappings.pmapping_group import PmappingDataframe
-from fastfusion.mapper.FFM._make_pmappings.make_pmappings import (
+from accelforge.mapper.FFM._join_pmappings.pmapping_group import PmappingDataframe
+from accelforge.mapper.FFM._make_pmappings.make_pmappings import (
     get_rank_variable_bounds_for_all_einsums,
 )
-from fastfusion._accelerated_imports import pd
+from accelforge._accelerated_imports import pd
 import joblib
-from fastfusion.mapper.FFM._join_pmappings.compatibility import Compatibility
-from fastfusion.mapper._simanneal2.tracking import EvaluationsScoreTracker
+from accelforge.mapper.FFM._join_pmappings.compatibility import Compatibility
+from accelforge.mapper._simanneal2.tracking import EvaluationsScoreTracker
 
 # Simulated annealing algorithm
 # -----------------------------

accelforge/_deprecate/compatibility_util.py

@@ -2,10 +2,10 @@
 
 from collections.abc import Iterable, Set
 
-from fastfusion.frontend.spec import Spec
-from fastfusion.frontend.workload import EinsumName, TensorName
-from fastfusion.mapper.FFM._join_pmappings.compatibility import Compatibility
-from fastfusion.mapper.FFM._join_pmappings.sim import SIM
+from accelforge.frontend.spec import Spec
+from accelforge.frontend.workload import EinsumName, TensorName
+from accelforge.mapper.FFM._join_pmappings.compatibility import Compatibility
+from accelforge.mapper.FFM._join_pmappings.sim import SIM
 
 
 DO_PRINT = False

accelforge/_deprecate/layerdeduplication/group_similar_einsums.py

@@ -2,10 +2,10 @@
 from collections.abc import Iterable
 from itertools import permutations, product
 
-from pytimeloop.bindings.looptree import LooptreeWorkload, LooptreeDependencyAnalyzer
+from pytimeloop.bindings._looptree import LooptreeWorkload, LooptreeDependencyAnalyzer
 
-from pytimeloop.looptree.mapping_utilities import get_intermediate_tensors
-from fastfusion.util._frozenset import fzs
+from pytimeloop._looptree.mapping_utilities import get_intermediate_tensors
+from accelforge.util._frozenset import fzs
 
 from .grouped_einsums import GroupOfSimilarEinsums, Id
 

accelforge/_deprecate/layerdeduplication/grouped_einsums.py

@@ -1,6 +1,6 @@
 from collections.abc import Iterable
 
-from bindings.looptree import LooptreeWorkload
+from bindings._looptree import LooptreeWorkload
 
 
 type Id = int

accelforge/_deprecate/mapping_filter_tags/ffmt.py

@@ -1,5 +1,5 @@
-from fastfusion.frontend.mapping import Loop, Temporal
-from fastfusion.mapper.FFM.deprecate_maybe.tags import Tags
+from accelforge.frontend.mapping import Loop, Temporal
+from accelforge.mapper.FFM.deprecate_maybe.tags import Tags
 
 from .util import get_fused_loops_per_tensor
 

accelforge/_deprecate/mapping_filter_tags/onesplit.py

@@ -1,5 +1,5 @@
-from fastfusion.mapper.FFM.deprecate_maybe.tags import Tags
-from fastfusion.mapper.FFM._join_pmappings.compatibility import Compatibility
+from accelforge.mapper.FFM.deprecate_maybe.tags import Tags
+from accelforge.mapper.FFM._join_pmappings.compatibility import Compatibility
 
 
 ONE_SPLIT = "ONE_SPLIT"

accelforge/_deprecate/mapping_filter_tags/util.py

@@ -1,4 +1,4 @@
-from fastfusion.frontend.mapping import Reservation, Loop, Mapping
+from accelforge.frontend.mapping import Reservation, Loop, Mapping
 
 
 def get_fused_loops_per_tensor(

accelforge/_deprecate/tags.py

@@ -11,7 +11,7 @@
 TagCompatibility as classes for the keys.
 """
 
-from fastfusion.util._frozenset import fzs
+from accelforge.util._frozenset import fzs
 
 
 class TagClass(fzs):

accelforge/_deprecate/viz/interactive.py

@@ -4,9 +4,9 @@
 from IPython.display import SVG, display
 import plotly.graph_objs as go
 from ipywidgets import Output, VBox, HBox
-from fastfusion._accelerated_imports import pd
-from fastfusion.mapper.FFM._join_pmappings.pmapping_dataframe import row2pmappings
-from fastfusion.frontend.mapping import Mapping
+from accelforge._accelerated_imports import pd
+from accelforge.mapper.FFM._join_pmappings.pmapping_dataframe import row2pmappings
+from accelforge.frontend.mapping import Mapping
 
 
 def make_mapping(row, einsum_names, rank_variable_bounds):

accelforge/_deprecate/viz/reservationtree.py

@@ -1,13 +1,13 @@
 from collections import defaultdict
 import pydot
 from typing import Any, Iterable
-from fastfusion.mapper.FFM._join_pmappings.pmapping_group import (
+from accelforge.mapper.FFM._join_pmappings.pmapping_group import (
     Compatibility,
     TensorReservation,
     Loop,
 )
-from fastfusion.util import _expfmt
-from fastfusion.mapper.FFM._join_pmappings.pmapping_dataframe import col2nameloop
+from accelforge.util import _expfmt
+from accelforge.mapper.FFM._join_pmappings.pmapping_dataframe import col2nameloop
 
 PYDOT_NODE_DEFAULTS = {
     "shape": "box",

accelforge/_deprecate/viz/ski_slope.py

@@ -1,9 +1,9 @@
 import matplotlib.axes as mpax
 import matplotlib.pyplot as plt
-from fastfusion._accelerated_imports import np
-from fastfusion._accelerated_imports import pd
+from accelforge._accelerated_imports import np
+from accelforge._accelerated_imports import pd
 
-from fastfusion.mapper.FFM._join_pmappings.pmapping_dataframe import PmappingDataframe
+from accelforge.mapper.FFM._join_pmappings.pmapping_dataframe import PmappingDataframe
 
 
 DATAFLOW_COLUMN = "dataflow"