integrations.kernels.autotune_callback
integrations.kernels.autotune_callback
Trainer callback for reporting selections from loaded Triton kernels.
Classes
| Name | Description |
|---|---|
| AutotuneReportCallback | Report selections from loaded Triton kernels. |
AutotuneReportCallback
integrations.kernels.autotune_callback.AutotuneReportCallback()Report selections from loaded Triton kernels.