Skip to content

Commit

Permalink
Numba backend cached property refactor (tests may compile slower sinc…
Browse files Browse the repository at this point in the history
…e JIT cache is now local to backend instance) (#1299)
  • Loading branch information
slayoo authored Jun 3, 2024
1 parent bac3407 commit ed5fb28
Show file tree
Hide file tree
Showing 45 changed files with 1,334 additions and 1,173 deletions.
4 changes: 2 additions & 2 deletions .github/workflows/tests+artifacts+pypi.yml
Original file line number Diff line number Diff line change
Expand Up @@ -131,7 +131,7 @@ jobs:
python-version: "3.8"
fail-fast: false
runs-on: ${{ matrix.platform }}
timeout-minutes: 45
timeout-minutes: ${{ startsWith(matrix.platform, 'windows-') && 45 || 30 }}
steps:
- uses: actions/checkout@v4.1.6
with:
Expand Down Expand Up @@ -216,7 +216,7 @@ jobs:
test-suite: [ "chemistry_freezing_isotopes", "condensation_a", "condensation_b", "coagulation", "breakup", "multi-process_a", "multi-process_b"]
fail-fast: false
runs-on: ${{ matrix.platform }}
timeout-minutes: 50
timeout-minutes: ${{ startsWith(matrix.platform, 'windows-') && 65 || 50 }}
steps:
- uses: actions/checkout@v4.1.6
with:
Expand Down
4 changes: 4 additions & 0 deletions PySDM/backends/impl_common/backend_methods.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,5 +9,9 @@ class BackendMethods:
def __init__(self):
if not hasattr(self, "formulae"):
self.formulae = None
if not hasattr(self, "formulae_flattened"):
self.formulae_flattened = None
if not hasattr(self, "Storage"):
self.Storage = None
if not hasattr(self, "default_jit_flags"):
self.default_jit_flags = {}
13 changes: 13 additions & 0 deletions PySDM/backends/impl_numba/methods/__init__.py
Original file line number Diff line number Diff line change
@@ -1 +1,14 @@
""" method classes of the CPU backend """

from .chemistry_methods import ChemistryMethods
from .collisions_methods import CollisionsMethods
from .condensation_methods import CondensationMethods
from .displacement_methods import DisplacementMethods
from .fragmentation_methods import FragmentationMethods
from .freezing_methods import FreezingMethods
from .index_methods import IndexMethods
from .isotope_methods import IsotopeMethods
from .moments_methods import MomentsMethods
from .pair_methods import PairMethods
from .physics_methods import PhysicsMethods
from .terminal_velocity_methods import TerminalVelocityMethods
Loading

0 comments on commit ed5fb28

Please sign in to comment.