__all__ = ["custom_add", "custom_trig"] from op_extension.custom_ops_lib import custom_add, custom_trig