Update to kernel registration

Egor-Krivov · Egor-Krivov · commit 4075a643d899 · 2025-07-14T11:58:32.000Z
diff --git a/bitsandbytes/backends/cuda/ops.py b/bitsandbytes/backends/cuda/ops.py
@@ -610,7 +610,7 @@ def _gemv_4bit_impl(
 }
 
 
-def optimizer_update_32bit(
+def _optimizer_update_32bit_impl(
     optimizer_name: str,
     g: torch.Tensor,
     p: torch.Tensor,
@@ -763,3 +763,4 @@ def _optimizer_update_8bit_blockwise_impl(
 
 
 register_kernel("bitsandbytes::optimizer_update_8bit_blockwise", "cuda")(_optimizer_update_8bit_blockwise_impl)
+register_kernel("bitsandbytes::optimizer_update_32bit", "cuda")(_optimizer_update_32bit_impl)

Original file line number	Diff line number	Diff line change
`@@ -610,7 +610,7 @@ def _gemv_4bit_impl(`
`610`	`610`	`}`
`611`	`611`
`612`	`612`
`613`		`-def optimizer_update_32bit(`
	`613`	`+def _optimizer_update_32bit_impl(`
`614`	`614`	`optimizer_name: str,`
`615`	`615`	`g: torch.Tensor,`
`616`	`616`	`p: torch.Tensor,`
`@@ -763,3 +763,4 @@ def _optimizer_update_8bit_blockwise_impl(`
`763`	`763`
`764`	`764`
`765`	`765`	`register_kernel("bitsandbytes::optimizer_update_8bit_blockwise", "cuda")(_optimizer_update_8bit_blockwise_impl)`
	`766`	`+register_kernel("bitsandbytes::optimizer_update_32bit", "cuda")(_optimizer_update_32bit_impl)`