fix for flaky tests in lora ops kernels w autotune (#3511) [skip ci]

* fix for flaky tests in lora ops kernels w autotune

* attempt 2 to fix
This commit is contained in:
Wing Lian
2026-03-19 01:18:47 -04:00
committed by GitHub
parent 5ef3f28340
commit f291ac029c
2 changed files with 41 additions and 10 deletions

View File

@@ -55,7 +55,7 @@ def _find_lora_ops_module() -> ModuleType | None:
``lora_ops`` and that has the ``_scatter2scatter_lora`` kernel
attribute — that is the runtime copy with populated caches.
"""
for name, module in sys.modules.items():
for name, module in list(sys.modules.items()):
if (
module is not None
and "lora_ops" in name