243a1b0276
Co-authored-by: Zhuohan Li <zhuohan@openai.com> Co-authored-by: Maratyszcza <marat@openai.com> Co-authored-by: Volodymyr Kyrylov <vol@wilab.org.ua>
7 lines
228 B
Python
7 lines
228 B
Python
from importlib import import_module as _im
|
|
|
|
# Load the compiled extension (gpt_oss.metal._metal)
|
|
_ext = _im(f"{__name__}._metal")
|
|
globals().update({k: v for k, v in _ext.__dict__.items() if not k.startswith("_")})
|
|
del _im, _ext
|