#pragma once

#define BNB_WARP_SIZE warpSize

// These are set based on current BNB support for CDNA 2 & RDNA 3. Update as needed for future archs
#define BNB_MAX_THREADS_PER_SM 2048
#define BNB_BF16_AVAILABLE true