fn supports_bf16()

in src/main.rs [317:325]

• 9 lines of code
• 2 McCabe index (conditional complexity)

fn supports_bf16(gpu: &Arc<CudaDevice>) -> anyhow::Result<bool> {
    Ok(
        gpu.attribute(sys::CUdevice_attribute_enum::CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MAJOR)?
            >= 8
            && gpu.attribute(
                sys::CUdevice_attribute_enum::CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MAJOR,
            )? >= 0,
    )
}