Skip to content

Commit

Permalink
cufinufft: update binsize to new default value
Browse files Browse the repository at this point in the history
  • Loading branch information
lgarrison committed Sep 3, 2024
1 parent 7c4fade commit db882ce
Showing 1 changed file with 9 additions and 6 deletions.
15 changes: 9 additions & 6 deletions src/jax_finufft/options.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,9 @@ class GpuMethod(IntEnum):

@dataclass(frozen=True)
class Opts:

# These correspond to the default cufinufft options
# set in vendor/finufft/src/cuda/cufinufft.cu
modeord: bool = False
chkbnds: bool = True
debug: DebugLevel = DebugLevel.Silent
Expand All @@ -59,12 +62,12 @@ class Opts:
gpu_upsampfac: float = 2.0
gpu_method: GpuMethod = 0
gpu_sort: bool = True
gpu_binsizex: int = -1
gpu_binsizey: int = -1
gpu_binsizez: int = -1
gpu_obinsizex: int = -1
gpu_obinsizey: int = -1
gpu_obinsizez: int = -1
gpu_binsizex: int = 0
gpu_binsizey: int = 0
gpu_binsizez: int = 0
gpu_obinsizex: int = 0
gpu_obinsizey: int = 0
gpu_obinsizez: int = 0
gpu_maxsubprobsize: int = 1024
gpu_kerevalmeth: bool = True
gpu_spreadinterponly: bool = False
Expand Down

0 comments on commit db882ce

Please sign in to comment.