def allow_cutlass_sm90_kernels() -> bool:
    """
    Whether the SM90 CUTLASS kernels should to be considered
    alongside the SM80 CUTLASS kernels on the CUDA arch 90
    (for the CUDA back-end of the GEMM ops). Default: False.
    """
    return (
        force_cutlass_sm90_kernels()
        or os.getenv("AIT_ALLOW_CUTLASS_SM90_KERNELS", "0") == "1"
    )
Whether the SM90 CUTLASS kernels should to be considered
alongside the SM80 CUTLASS kernels on the CUDA arch 90
(for the CUDA back-end of the GEMM ops). Default: False.
python/aitemplate/utils/environ.py