integrations.liger.utils
integrations.liger.utils
utils to patch liger kernel ops to disable torch.compile
Functions
| Name | Description |
|---|---|
| patch_with_compile_disable | Patch a function in a module by wrapping it with torch.compile.disable |
patch_with_compile_disable
integrations.liger.utils.patch_with_compile_disable(module, function_name)Patch a function in a module by wrapping it with torch.compile.disable
Parameters
| Name | Type | Description | Default |
|---|---|---|---|
| module | The module containing the function to patch | required | |
| function_name | The name of the function to patch | required |