Skip to content

Latest commit

 

History

History

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 

Custom user structure examples

This folder contains examples of custom user structures in cupy.RawKernel (see https://docs.cupy.dev/en/stable/tutorial/kernel.html for corresponding documentation).

This folder provides three scripts ranked by increasing complexity:

  1. builtins_vectors.py shows how to use CUDA builtin vectors such as float4 both as scalar parameter (pass by value from host) and array parameter in RawKernels.
  2. packed_matrix.py demonstrates how to create and use templated packed structures in RawModules.
  3. complex_struct.py illustrates the possibility to recursively build complex NumPy dtypes matching device structure memory layout.

All examples can be run as simple python scripts: python3.x example_name.py.