def infer_plugin_from_build_config()

in src/optimum/nvidia/export/converter.py [0:0]

• 6 lines of code
• 3 McCabe index (conditional complexity)

def infer_plugin_from_build_config(config: "BuildConfig") -> "BuildConfig":
    if is_post_ampere():
        # Required for Chunk Context
        LOGGER.debug("Enabling Paged Context FMHA plugin")
        config.plugin_config.update_from_dict({"use_paged_context_fmha": True})

    config.plugin_config.update_from_dict({"enable_xqa": False})
    return config