# Copyright 2024 The IREE Authors
#
# Licensed under the Apache License v2.0 with LLVM Exceptions.
# See https://llvm.org/LICENSE.txt for license information.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception


class VmfbManager:
    sdxl_clip_cpu_vmfb = None
    sdxl_vae_cpu_vmfb = None
    sdxl_unet_fp16_cpu_vmfb = None
    sdxl_unet_fp16_cpu_pipeline_vmfb = None
    sdxl_scheduler_cpu_vmfb = None
    sdxl_clip_rocm_vmfb = None
    sdxl_vae_rocm_vmfb = None
    sdxl_unet_fp16_rocm_vmfb = None
    sdxl_punet_int8_fp16_rocm_vmfb = None
    sdxl_punet_int8_fp8_rocm_vmfb = None
    sdxl_unet_fp16_rocm_pipeline_vmfb = None
    sdxl_scheduler_rocm_vmfb = None
    sd3_clip_cpu_vmfb = None
    sd3_vae_cpu_vmfb = None
    sd3_mmdit_cpu_vmfb = None
    sd3_clip_rocm_vmfb = None
    sd3_vae_rocm_vmfb = None
    sd3_mmdit_rocm_vmfb = None