kernel-craft 0.1.0__py3-none-any.whl

kernel_craft-0.1.0.dist-info/METADATA

@@ -0,0 +1,108 @@
+Metadata-Version: 2.4
+Name: kernel-craft
+Version: 0.1.0
+Summary: CUDA kernels for machine learning systems optimization
+Author-email: kernel-craft contributors <contact@example.com>
+License-Expression: MIT
+Project-URL: Homepage, https://github.com/anomalyco/kernel-craft
+Keywords: cuda,gpu,convolution,machine-learning,deep-learning
+Classifier: Development Status :: 3 - Alpha
+Classifier: Programming Language :: Python :: 3
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Environment :: GPU :: NVIDIA CUDA
+Classifier: Operating System :: POSIX :: Linux
+Requires-Python: <3.13,>=3.11
+Description-Content-Type: text/markdown
+Requires-Dist: numpy>=1.20
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0; extra == "dev"
+Requires-Dist: twine>=4.0.0; extra == "dev"
+Requires-Dist: build>=1.0.0; extra == "dev"
+
+# kernel-craft Python API
+
+CUDA convolution kernels exposed to Python with numpy and PyTorch support.
+
+## Installation
+
+Build from source with CMake:
+
+```bash
+cd /path/to/kernel-craft
+mkdir build && cd build
+cmake ..
+make kernel_craft_python
+```
+
+The module will be at `src/python/build/kernel_craft_python.cpython-*.so`.
+
+## Usage
+
+```python
+import sys
+sys.path.insert(0, 'src/python/build')
+
+import kernel_craft_python as kc
+import numpy as np
+
+# Input: 2D float32 numpy array
+input = np.random.randn(256, 256).astype(np.float32)
+kernel = np.random.randn(3, 3).astype(np.float32)
+
+# Naive convolution
+out = kc.conv_naive(input, kernel)  # -> np.ndarray
+
+# Tiled convolution with configurable tile size
+out = kc.conv_tiled(input, kernel, tile_w=8, tile_h=8)  # -> np.ndarray
+```
+
+## Version
+
+```python
+import kernel_craft_python as kc
+print(kc.__version__)  # "0.1.0"
+```
+
+## PyTorch Tensors
+
+```python
+import torch
+import kernel_craft_python as kc
+
+# Input: 2D float32 PyTorch tensor on CUDA
+input = torch.rand(256, 256, dtype=torch.float32, device='cuda')
+kernel = torch.rand(3, 3, dtype=torch.float32, device='cuda')
+
+# Naive convolution
+out = kc.conv_naive(input, kernel)  # -> torch.Tensor on GPU
+
+# Tiled convolution
+out = kc.conv_tiled(input, kernel, tile_w=16, tile_h=16)  # -> torch.Tensor on GPU
+```
+
+## API Reference
+
+| Function | Input Type | Output Type |
+|----------|-----------|--------------|
+| `conv_naive(input, kernel)` | np.ndarray or Tensor | np.ndarray or Tensor |
+| `conv_tiled(input, kernel, tile_w, tile_h)` | np.ndarray or Tensor | np.ndarray or Tensor |
+
+### Parameters
+
+- `input`: Input image (2D, float32)
+- `kernel`: Convolution kernel (2D, float32, odd dimension)
+- `tile_w`: Tile width for tiled convolution (default: 8)
+- `tile_h`: Tile height for tiled convolution (default: 8)
+
+### Supported Tile Sizes
+
+- 8x8 (default, best overall performance)
+- 16x16
+- 32x32
+
+### Error Handling
+
+All functions raise `RuntimeError` with descriptive messages for:
+- Invalid input dimensions (must be 2D)
+- Invalid kernel dimensions (must be 2D, square, odd-sized)
+- Invalid dtype (must be float32)

kernel_craft-0.1.0.dist-info/RECORD

@@ -0,0 +1,5 @@
+kernel_craft_python/kernel_craft_python.cpython-312-x86_64-linux-gnu.so,sha256=OzoOlFBKuu9PYPNCAAVFT4eYVc5eAD0iiD4LUb-XI9g,384984
+kernel_craft-0.1.0.dist-info/METADATA,sha256=Ac68AWPDnUo8AtwbN5DwLXg_Lf5xAnIshLZh7W5G6c4,2957
+kernel_craft-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+kernel_craft-0.1.0.dist-info/top_level.txt,sha256=RziAgd9_MYWWvPDd2hwcRwUZP2a2BL9rnIVF6eYOz3o,20
+kernel_craft-0.1.0.dist-info/RECORD,,

kernel_craft-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

kernel_craft-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+kernel_craft_python