Environment Variables

PyKokkos behavior can be controlled through the following environment variables:

Variable	Value	Description
PK_FORMAT	any (presence triggers)	Runs clang-format on intermediate C++ sources.
PK_FUSION	"naive"	Enables automatic kernel fusion. See Kernel Fusion.
PK_KOKKOS_LIB_PATH	path to directory	Overrides search path for the compiled pykokkos-base lib/ or lib64/ directory.
PK_KOKKOS_INTERFACE	Kokkos version string	Selects a specific version of the Kokkos interface to use when multiple versions are available.

Usage Examples

Enable C++ code formatting for generated kernels:

export PK_FORMAT=1

Enable naive kernel fusion:

export PK_FUSION="naive"

Specify a custom Kokkos library path:

export PK_KOKKOS_LIB_PATH="/path/to/pykokkos-base/lib"