Skip to content

Commit

Permalink
readme: fix link and wording
Browse files Browse the repository at this point in the history
  • Loading branch information
lgarrison authored Dec 8, 2024
1 parent f4b0aa2 commit fa53f25
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ All of these functions support forward, reverse, and higher-order differentiatio
as well as batching using `vmap`.

> [!NOTE]
> The GPU backend does not currently support 1D (#125).
> The GPU backend does not currently support 1D ([#125](https://github.com/flatironinstitute/jax-finufft/issues/125)).
## Installation

Expand Down Expand Up @@ -139,7 +139,7 @@ There are several important CMake variables that control aspects of the jax-finu

- **`JAX_FINUFFT_USE_CUDA`** [disabled by default]: build with GPU support
- **`CMAKE_CUDA_ARCHITECTURES`** [default `native`]: the target GPU architecture. `native` means the GPU arch of the build system.
- **`FINUFFT_ARCH_FLAGS`** [default `-march=native`]: the target CPU architecture. `native` means the CPU arch of the build system.
- **`FINUFFT_ARCH_FLAGS`** [default `-march=native`]: the target CPU architecture. The default is the native CPU arch of the build system.

Each of these can be set as `-Ccmake.define.NAME=VALUE` arguments to `pip install`. For example,
to build with GPU support from the repo root, run:
Expand Down

0 comments on commit fa53f25

Please sign in to comment.