torch-bessel 0.0.5__cp311-cp311-macosx_11_0_arm64.whl → 0.0.6__cp311-cp311-macosx_11_0_arm64.whl

{torch_bessel-0.0.5.dist-info → torch_bessel-0.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: torch_bessel
-Version: 0.0.5
+Version: 0.0.6
 Summary: PyTorch extension package for Bessel functions with arbitrary real order and complex inputs
 Home-page: https://github.com/hchau630/torch-bessel
 Author: Ho Yin Chau
@@ -12,6 +12,7 @@ Dynamic: author
 Dynamic: description
 Dynamic: description-content-type
 Dynamic: home-page
+Dynamic: license-file
 Dynamic: requires-dist
 Dynamic: requires-python
 Dynamic: summary
@@ -35,7 +36,7 @@ torch_bessel.ops.modified_bessel_k0(z)
 ```
 
 # Implemented functions
-- `modified_bessel_k0`: Same as `torch.special.modified_bessel_k0`, but also handles backpropagation and complex inputs on cpu and cuda. Correctness is guaranteed on the right-half complex plane. On the left-half complex plane, function output appears mostly correct, but with small numerical errors for certain inputs. On the negative real line, output is NaN.
+- `modified_bessel_k0`: Same as `torch.special.modified_bessel_k0`, but also handles backpropagation and complex inputs on cpu and cuda. Correctness is guaranteed on the right-half complex plane for double types, and almost guaranteed for float types, though it appears there are a very small handful of inputs which result in NaNs which needs to be fixed. On cuda, `torch.chalf` inputs are also supported, though the underlying cuda kernel just upcasts `chalf` to `cfloat` (note that this uses no extra GPU memory, as opposed to manually casting torch.chalf to torch.cfloat before calling `modified_bessel_k0` which doubles the GPU memory used). On the left-half complex plane, function output appears mostly correct, but with small numerical errors for certain inputs. On the negative real line, output is NaN.
 
 # WIP
 - `modified_bessel_kv`: Analogue of `scipy.special.kv`.

torch_bessel-0.0.6.dist-info/RECORD

@@ -0,0 +1,10 @@
+benchmarks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+benchmarks/benchmarks.py,sha256=LZDGAlKEWwo-VCenNTGn98bJzj9FSlQbFK5pPXRvDyI,3319
+torch_bessel/_C.cpython-311-darwin.so,sha256=_Lyk6INVRLDmaAovlBKqHDkEK6h-fNVqB7Pc_PeqvKs,231224
+torch_bessel/__init__.py,sha256=oohbWz8vxekl7kqDNSWqDH3ORabf9-Tug1KJryKw51A,230
+torch_bessel/ops.py,sha256=P4qo1GN6XExuKfI061dsZMpAPuimn6gYt-PwRmR4_Fk,3584
+torch_bessel-0.0.6.dist-info/licenses/LICENSE,sha256=do0DI6wu4mF3VXnEXXPYZqVEatoRSSamgz9t80wU7_o,1068
+torch_bessel-0.0.6.dist-info/METADATA,sha256=uViUBBZq-V6HuSKg2N4NMiWJYtJwUxmLxkB0dtTVzS4,2034
+torch_bessel-0.0.6.dist-info/WHEEL,sha256=qxQkdhERtGxJzqnVOBnucx1aUmU2n3HmuzYdln_LyOw,109
+torch_bessel-0.0.6.dist-info/top_level.txt,sha256=xmyVjWSQ91kX-v8KCzl6wDwfAmbdZNWsP2EH9b9BccQ,24
+torch_bessel-0.0.6.dist-info/RECORD,,

{torch_bessel-0.0.5.dist-info → torch_bessel-0.0.6.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (76.0.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: false
 Tag: cp311-cp311-macosx_11_0_arm64
 

torch_bessel-0.0.5.dist-info/RECORD

@@ -1,10 +0,0 @@
-benchmarks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-benchmarks/benchmarks.py,sha256=LZDGAlKEWwo-VCenNTGn98bJzj9FSlQbFK5pPXRvDyI,3319
-torch_bessel/_C.cpython-311-darwin.so,sha256=8vYRQgQ465bcJE3ENxvjLx_hKY_AX5ehFZbbFb1OiaI,231224
-torch_bessel/__init__.py,sha256=oohbWz8vxekl7kqDNSWqDH3ORabf9-Tug1KJryKw51A,230
-torch_bessel/ops.py,sha256=P4qo1GN6XExuKfI061dsZMpAPuimn6gYt-PwRmR4_Fk,3584
-torch_bessel-0.0.5.dist-info/LICENSE,sha256=do0DI6wu4mF3VXnEXXPYZqVEatoRSSamgz9t80wU7_o,1068
-torch_bessel-0.0.5.dist-info/METADATA,sha256=LX6X8j53vJCYKaXLfjlk3X_RwRYHlkoQVYC7CSJJ1vc,1561
-torch_bessel-0.0.5.dist-info/WHEEL,sha256=uZm-0PiOS8tRdZxJBJgtzs5SssLMMKtm3LCPIWseOMM,109
-torch_bessel-0.0.5.dist-info/top_level.txt,sha256=xmyVjWSQ91kX-v8KCzl6wDwfAmbdZNWsP2EH9b9BccQ,24
-torch_bessel-0.0.5.dist-info/RECORD,,