fix(library): use name in define

huggingface · Feb 8, 2024 · 4bffe4e · 4bffe4e
1 parent f634e07
commit 4bffe4e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/quanto/library/ops.py b/quanto/library/ops.py
@@ -38,7 +38,7 @@ def define(name, schema):
         torch.library.define(f"{libname}::{name}", schema)
 
     # Provide the inplementation for all dispatch key in the main library
-    @torch.library.impl("quanto::unpack", "default")
+    @torch.library.impl(f"quanto::{name}", "default")
     def impl(*args, **kwargs):
         if _ext_enabled:
             return getattr(torch.ops.quanto_ext, name)(*args, **kwargs)