NVIDIA / NVIDIA/cccl

cuda.compute: Expose a way to pass additional compilation options to NVRTC

Open
#6,873 9 comments 2 reactions 1 assignee Assigned to @Copilot View on GitHub
Dominant language
C++
Stars
2.5k
Forks
486
Avg merge
2d 6h
Merged PRs (30d)
295

Description

https://docs.nvidia.com/cuda/nvrtc/index.html#supported-compile-options

It would be useful to be able to pass additional compiler flags to NVRTC to control e.g., floating point math.

The CCCL.c layer already provides this facility, we just need to plumb it through to the user.

The API could look something like:

```python
set_nvrtc_compile_options({"fmad": "true"})
```

Or on a per algorithm basis:

```python
reducer = make_reduce_into(d_in, d_out, op, h_init, num_items, nvrtc_compile_options={"fmad": "true"})
```

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.