描述

🚀 Feature

We should be able to: (1) Understand which op is used for FW/BW (via additional logging enabled with an env variable?) (2) Possibly disable some ops entirely with env variable (like CUDA_VISIBLE_DEVICES but something like XFORMERS_MEMEFF_FW_OPERATORS=...)

貢獻者指南

技術棧: pythonpytorch
領域: performancemachine learning
議題類型: feature
難度: 3
預計時間: half day
活動狀態: stale
清晰度: mostly clear
前置要求: PythonPyTorchEnvironment variables
新手友善度: 35
研究方向: Review the xformers codebase, particularly the modules related to memory efficient operations (e.g., xformers/ops/). Examine existing environment variable usage patterns. Consider adding logging to track which ops are used, and a mechanism to disable specific ops via an env variable, similar to CUDA VISIBLE DEVICES. The implementation should be in the core module.