Skip to content

Commit

Permalink
Added nox-poetry to GitHub test Action
Browse files Browse the repository at this point in the history
  • Loading branch information
fjwillemsen committed Sep 26, 2023
1 parent b08950a commit 2e8edad
Show file tree
Hide file tree
Showing 3 changed files with 9 additions and 7 deletions.
4 changes: 3 additions & 1 deletion .github/workflows/test-python-package.yml
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,9 @@ jobs:
- uses: actions/checkout@v4
- name: Run tests with Nox
uses: fjwillemsen/[email protected]
- run: nox -- skip-gpu
- run: |
pip install nox-poetry
nox -- skip-gpu
# - name: Upload Coverage report to CodeCov
# uses: codecov/codecov-action@v3
# with:
Expand Down
2 changes: 1 addition & 1 deletion noxfile.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
import nox

# from nox import Session, session
from nox_poetry import Session, session # nox_poetry is a better option, but <=1.0.3 has a bug with filename-URLs
from nox_poetry import Session, session

python_versions_to_test = ["3.8", "3.9", "3.10", "3.11"]
nox.options.stop_on_first_error = True
Expand Down
10 changes: 5 additions & 5 deletions test/test_searchspace.py
Original file line number Diff line number Diff line change
Expand Up @@ -415,11 +415,11 @@ def test_full_searchspace(compare_against_bruteforce=False):
tune_params["blocks_per_sm"] = [0,1,2,3,4]

restrictions = [
"block_size_x*block_size_y >= 32", # doesn't work on its own (3719987 vs 7800000)
"temporal_tiling_factor % loop_unroll_factor_t == 0", # works fine on its own
f"block_size_x*block_size_y <= {dev['max_threads']}", # doesn't work on its own (1688843 vs 9300000)
f"(block_size_x*tile_size_x + temporal_tiling_factor * 2) * (block_size_y*tile_size_y + temporal_tiling_factor * 2) * (2+sh_power) * 4 <= {dev['max_shared_memory_per_block']}", # works fine on its own
f"blocks_per_sm == 0 or (((block_size_x*tile_size_x + temporal_tiling_factor * 2) * (block_size_y*tile_size_y + temporal_tiling_factor * 2) * (2+sh_power) * 4) * blocks_per_sm <= {dev['max_shared_memory']})" # works fine on its own
"block_size_x*block_size_y >= 32",
"temporal_tiling_factor % loop_unroll_factor_t == 0",
f"block_size_x*block_size_y <= {dev['max_threads']}",
f"(block_size_x*tile_size_x + temporal_tiling_factor * 2) * (block_size_y*tile_size_y + temporal_tiling_factor * 2) * (2+sh_power) * 4 <= {dev['max_shared_memory_per_block']}",
f"blocks_per_sm == 0 or (((block_size_x*tile_size_x + temporal_tiling_factor * 2) * (block_size_y*tile_size_y + temporal_tiling_factor * 2) * (2+sh_power) * 4) * blocks_per_sm <= {dev['max_shared_memory']})"
]

# build the searchspace
Expand Down

0 comments on commit 2e8edad

Please sign in to comment.