pystencils.backend.platforms.CudaPlatform

class pystencils.backend.platforms.CudaPlatform(ctx, *, assume_warp_aligned_block_size=False, warp_size=None)

Platform for the CUDA GPU target.

Parameters:

• ctx (KernelCreationContext)

• assume_warp_aligned_block_size (bool)

• warp_size (int | None)

property required_headers: set[str]

Set of header files that must be included at the point of definition of a kernel running on this platform.