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