mlx/backend/cuda/device/config.h

// Copyright © 2025 Apple Inc.

// This file is used by both CUDA kernel code and host-only C++ code.

#pragma once

// The maximum dimensions of shape/strides passed as kernel parameters.
#define MAX_NDIM 10

// All existing NVIDIA hardware has a fixed 32 warp size. Though a built-in
// warpSize variable exists, using it would prevent compile-time optimizations.
#define WARP_SIZE 32
CUDA backend: compile (#2276) * CUDA backend: compile * Rename kernels/ to device/ 2025-06-13 09:08:39 +09:00			`// Copyright © 2025 Apple Inc.`

			`// This file is used by both CUDA kernel code and host-only C++ code.`

			`#pragma once`

			`// The maximum dimensions of shape/strides passed as kernel parameters.`
divmod, partition, sort fixes (#2302) 2025-06-16 18:49:32 -07:00			`#define MAX_NDIM 10`
CUDA backend: compile (#2276) * CUDA backend: compile * Rename kernels/ to device/ 2025-06-13 09:08:39 +09:00
			`// All existing NVIDIA hardware has a fixed 32 warp size. Though a built-in`
			`// warpSize variable exists, using it would prevent compile-time optimizations.`
			`#define WARP_SIZE 32`