kernels 0.3.2__tar.gz → 0.4.0__tar.gz

{kernels-0.3.2 → kernels-0.4.0}/PKG-INFO

@@ -1,13 +1,13 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.1
 Name: kernels
-Version: 0.3.2
+Version: 0.4.0
 Summary: Download compute kernels
 Author-email: OlivierDehaene <olivier@huggingface.co>, Daniel de Kok <daniel@huggingface.co>, David Holtz <david@huggingface.co>, Nicolas Patry <nicolas@huggingface.co>
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
-Requires-Dist: huggingface-hub>=0.26.3
-Requires-Dist: packaging>=24.2
-Requires-Dist: tomli>=2.0.1; python_version < "3.11"
+Requires-Dist: huggingface_hub<1.0,>=0.26.0
+Requires-Dist: packaging>=20.0
+Requires-Dist: tomli>=2.0; python_version < "3.11"
 Provides-Extra: torch
 Requires-Dist: torch; extra == "torch"
 

{kernels-0.3.2 → kernels-0.4.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "kernels"
-version = "0.3.2"
+version = "0.4.0"
 description = "Download compute kernels"
 authors = [
   { name = "OlivierDehaene", email = "olivier@huggingface.co" },
@@ -11,9 +11,9 @@ authors = [
 readme = "README.md"
 requires-python = ">= 3.9"
 dependencies = [
-  "huggingface-hub>=0.26.3",
-  "packaging>=24.2",
-  "tomli>=2.0.1; python_version<'3.11'",
+  "huggingface_hub>=0.26.0,<1.0",
+  "packaging>=20.0",
+  "tomli>=2.0; python_version<'3.11'",
 ]
 
 [build-system]

{kernels-0.3.2 → kernels-0.4.0}/src/kernels/layer.py

@@ -54,11 +54,26 @@ _KERNEL_MAPPING: ContextVar[Dict[str, Dict[Device, LayerRepository]]] = ContextV
 )
 
 
-def use_kernel_mapping(mapping: Dict[str, Dict[Union[Device, str], LayerRepository]]):
+def use_kernel_mapping(
+    mapping: Dict[str, Dict[Union[Device, str], LayerRepository]],
+    *,
+    inherit_mapping: bool = True,
+):
+    """
+    Context manager that sets a mapping for a duration of the context.
+
+    When `inherit_mapping` is set to `True` the current mapping will be
+    extended by `mapping` inside the context. If it is `False`, only
+    `mapping` is used inside the context.
+    """
+
     class ContextManager:
         def __enter__(self):
             # Mappings always stack on previous mappings.
-            self.token = _KERNEL_MAPPING.set(deepcopy(_KERNEL_MAPPING.get()))
+            if inherit_mapping:
+                self.token = _KERNEL_MAPPING.set(deepcopy(_KERNEL_MAPPING.get()))
+            else:
+                self.token = _KERNEL_MAPPING.set({})
             register_kernel_mapping(mapping)
 
         def __exit__(self, exc_type, exc_value, traceback):

{kernels-0.3.2 → kernels-0.4.0}/src/kernels.egg-info/PKG-INFO

@@ -1,13 +1,13 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.1
 Name: kernels
-Version: 0.3.2
+Version: 0.4.0
 Summary: Download compute kernels
 Author-email: OlivierDehaene <olivier@huggingface.co>, Daniel de Kok <daniel@huggingface.co>, David Holtz <david@huggingface.co>, Nicolas Patry <nicolas@huggingface.co>
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
-Requires-Dist: huggingface-hub>=0.26.3
-Requires-Dist: packaging>=24.2
-Requires-Dist: tomli>=2.0.1; python_version < "3.11"
+Requires-Dist: huggingface_hub<1.0,>=0.26.0
+Requires-Dist: packaging>=20.0
+Requires-Dist: tomli>=2.0; python_version < "3.11"
 Provides-Extra: torch
 Requires-Dist: torch; extra == "torch"
 

kernels-0.4.0/src/kernels.egg-info/requires.txt

@@ -0,0 +1,8 @@
+huggingface_hub<1.0,>=0.26.0
+packaging>=20.0
+
+[:python_version < "3.11"]
+tomli>=2.0
+
+[torch]
+torch

{kernels-0.3.2 → kernels-0.4.0}/tests/test_layer.py

@@ -152,6 +152,25 @@ def test_mapping_contexts():
             == "kernels-community/activation"
         )
 
+        with use_kernel_mapping(extra_mapping2, inherit_mapping=False):
+            assert set(_KERNEL_MAPPING.get().keys()) == {
+                "SiluAndMul",
+            }
+            assert (
+                _KERNEL_MAPPING.get()["SiluAndMul"][Device(type="cuda")].repo_id
+                == "kernels-community/non-existing"
+            )
+
+        assert set(_KERNEL_MAPPING.get().keys()) == {
+            "SiluAndMul",
+            "SiluAndMulStringDevice",
+            "TestKernel",
+        }
+        assert (
+            _KERNEL_MAPPING.get()["SiluAndMul"][Device(type="cuda")].repo_id
+            == "kernels-community/activation"
+        )
+
     assert set(_KERNEL_MAPPING.get().keys()) == {
         "SiluAndMul",
         "SiluAndMulStringDevice",

kernels-0.3.2/src/kernels.egg-info/requires.txt

@@ -1,8 +0,0 @@
-huggingface-hub>=0.26.3
-packaging>=24.2
-
-[:python_version < "3.11"]
-tomli>=2.0.1
-
-[torch]
-torch