diff --git a/src/cunumpy/_cuda_kernel.py b/src/cunumpy/_cuda_kernel.py index 1ca92e2..1c691d1 100644 --- a/src/cunumpy/_cuda_kernel.py +++ b/src/cunumpy/_cuda_kernel.py @@ -326,10 +326,6 @@ def _includes(source: str) -> list[tuple[str, bool]]: ] -def _quoted_includes(source: str) -> list[str]: - return [name for name, quoted in _includes(source) if quoted] - - def resolve_includes( source: str, include_dirs: Iterable[str | Path] = (), @@ -949,7 +945,7 @@ def _header_source( if any(f.view_ndim is not None for s in structs for f in s.fields): includes.insert(0, _ARRAY_VIEW_INCLUDE) lines = [ - "// Generated by cunumpy.cuda.CudaStruct from the Python definition; do not edit.", + "// Generated by cunumpy.arguments.CudaStruct from the Python definition; do not edit.", f"#ifndef {guard}", f"#define {guard}", "", diff --git a/tests/unit/test_cuda_kernel.py b/tests/unit/test_cuda_kernel.py index 15eedc6..011fcf2 100644 --- a/tests/unit/test_cuda_kernel.py +++ b/tests/unit/test_cuda_kernel.py @@ -1111,7 +1111,7 @@ def test_to_header(tmp_path): struct = CudaStruct.from_signature(MarkerArguments.__init__, "MarkerArgs") header = struct.to_header() assert header == ( - "// Generated by cunumpy.cuda.CudaStruct from the Python definition; do not edit.\n" + "// Generated by cunumpy.arguments.CudaStruct from the Python definition; do not edit.\n" "#ifndef MARKERARGS_CUH\n" "#define MARKERARGS_CUH\n" "\n" @@ -1151,7 +1151,7 @@ def test_write_cuda_header(tmp_path): ) assert path.read_text() == header assert header.startswith( - "// Generated by cunumpy.cuda.CudaStruct from the Python definition; do not edit.\n" + "// Generated by cunumpy.arguments.CudaStruct from the Python definition; do not edit.\n" "#ifndef PUSHER_ARGS_CUH\n#define PUSHER_ARGS_CUH\n\n" '#include "cunumpy/array_view.cuh"\n#include \n\n', )