Merge pull request #54 from HERA-Team/fix-block-size

fix: better block/grid size determination

Author: steven-murray

.github/workflows/publish-to-pypi.yaml

@@ -5,7 +5,7 @@ on: push
 jobs:
   build-n-publish:
     name: Deploy
-    runs-on: ubuntu-18.04
+    runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@master
       # https://github.com/ansible/pylibssh/blob/1e7b17f/.github/workflows/build-test-n-publish.yml#L146-L151
@@ -23,7 +23,6 @@ jobs:
         run: |
           python -m pip install build
           python -m pip install setuptools_scm setuptools>42
-          python setup.py --version
       - name: Create setuptools_scm env variable
         if: startsWith(github.ref, 'refs/tags/') != true
         shell: bash

.gitignore

@@ -53,3 +53,4 @@ docs/_autosummary/*
 full-stats-*.txt
 stats-*.pkl
 summary-stats-*.pkl
+.hypothesis/*

gpu-analysis.nvprof

@@ -59,6 +59,7 @@ profile =
     line-profiler
     pyuvsim>=1.2.5
 test =
+    hypothesis
     ipython
     matplotlib
     pyradiosky
@@ -109,6 +110,7 @@ ignore =
 per-file-ignores =
     src/vis_cpu/cli.py: T201,T001
     tests/*.py: T201,T001,E402
+    setup.py: D100
 max-line-length = 88
 max-complexity = 25
 rst-roles =

setup.cfg

@@ -0,0 +1,3 @@
+from setuptools import setup
+
+setup()

setup.py

@@ -6,6 +6,14 @@ def no_op(fnc):
     return fnc
 
 
+def ceildiv(a: int, b: int) -> int:
+    """Ceiling division for integers.
+
+    From https://stackoverflow.com/a/17511341/1467820
+    """
+    return -(a // -b)
+
+
 def human_readable_size(size, decimal_places=2, indicate_sign=False):
     """Get a human-readable data size.
 

src/vis_cpu/_utils.py

@@ -230,7 +230,7 @@ def get_summary_stats(line_data, total_time, ids):
         if not assoc_lines:
             raise RuntimeError(
                 f"Could not find any lines for {thing} satisfying '{lines}'. "
-                f"Possible lines: {' | '.join(list(line_data.keys()))}"
+                "Possible lines:\n" + "\n".join(list(line_data.keys()))
             )
 
         # save (hits, time, time/hits, percent, nlines)