torch-bessel 0.0.5__cp312-cp312-macosx_11_0_arm64.whl → 0.0.6__cp312-cp312-macosx_11_0_arm64.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- torch_bessel/_C.cpython-312-darwin.so +0 -0
- {torch_bessel-0.0.5.dist-info → torch_bessel-0.0.6.dist-info}/METADATA +4 -3
- torch_bessel-0.0.6.dist-info/RECORD +10 -0
- {torch_bessel-0.0.5.dist-info → torch_bessel-0.0.6.dist-info}/WHEEL +1 -1
- torch_bessel-0.0.5.dist-info/RECORD +0 -10
- {torch_bessel-0.0.5.dist-info → torch_bessel-0.0.6.dist-info/licenses}/LICENSE +0 -0
- {torch_bessel-0.0.5.dist-info → torch_bessel-0.0.6.dist-info}/top_level.txt +0 -0
Binary file
|
@@ -1,6 +1,6 @@
|
|
1
|
-
Metadata-Version: 2.
|
1
|
+
Metadata-Version: 2.4
|
2
2
|
Name: torch_bessel
|
3
|
-
Version: 0.0.
|
3
|
+
Version: 0.0.6
|
4
4
|
Summary: PyTorch extension package for Bessel functions with arbitrary real order and complex inputs
|
5
5
|
Home-page: https://github.com/hchau630/torch-bessel
|
6
6
|
Author: Ho Yin Chau
|
@@ -12,6 +12,7 @@ Dynamic: author
|
|
12
12
|
Dynamic: description
|
13
13
|
Dynamic: description-content-type
|
14
14
|
Dynamic: home-page
|
15
|
+
Dynamic: license-file
|
15
16
|
Dynamic: requires-dist
|
16
17
|
Dynamic: requires-python
|
17
18
|
Dynamic: summary
|
@@ -35,7 +36,7 @@ torch_bessel.ops.modified_bessel_k0(z)
|
|
35
36
|
```
|
36
37
|
|
37
38
|
# Implemented functions
|
38
|
-
- `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.
|
39
|
+
- `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.
|
39
40
|
|
40
41
|
# WIP
|
41
42
|
- `modified_bessel_kv`: Analogue of `scipy.special.kv`.
|
@@ -0,0 +1,10 @@
|
|
1
|
+
benchmarks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
2
|
+
benchmarks/benchmarks.py,sha256=LZDGAlKEWwo-VCenNTGn98bJzj9FSlQbFK5pPXRvDyI,3319
|
3
|
+
torch_bessel/_C.cpython-312-darwin.so,sha256=YZJPORMvIiAa-JColLS3KoyD0HI60ow1aRyuzjlsif8,232280
|
4
|
+
torch_bessel/__init__.py,sha256=oohbWz8vxekl7kqDNSWqDH3ORabf9-Tug1KJryKw51A,230
|
5
|
+
torch_bessel/ops.py,sha256=P4qo1GN6XExuKfI061dsZMpAPuimn6gYt-PwRmR4_Fk,3584
|
6
|
+
torch_bessel-0.0.6.dist-info/licenses/LICENSE,sha256=do0DI6wu4mF3VXnEXXPYZqVEatoRSSamgz9t80wU7_o,1068
|
7
|
+
torch_bessel-0.0.6.dist-info/METADATA,sha256=uViUBBZq-V6HuSKg2N4NMiWJYtJwUxmLxkB0dtTVzS4,2034
|
8
|
+
torch_bessel-0.0.6.dist-info/WHEEL,sha256=CltXN3lQvXbHxKDtiDwW0RNzF8s2WyBuPbOAX_ZeQlA,109
|
9
|
+
torch_bessel-0.0.6.dist-info/top_level.txt,sha256=xmyVjWSQ91kX-v8KCzl6wDwfAmbdZNWsP2EH9b9BccQ,24
|
10
|
+
torch_bessel-0.0.6.dist-info/RECORD,,
|
@@ -1,10 +0,0 @@
|
|
1
|
-
benchmarks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
2
|
-
benchmarks/benchmarks.py,sha256=LZDGAlKEWwo-VCenNTGn98bJzj9FSlQbFK5pPXRvDyI,3319
|
3
|
-
torch_bessel/_C.cpython-312-darwin.so,sha256=ARpLpmh0MCoSGplMTVRxOWtT2zCGI8DbJDY7satv5NM,232280
|
4
|
-
torch_bessel/__init__.py,sha256=oohbWz8vxekl7kqDNSWqDH3ORabf9-Tug1KJryKw51A,230
|
5
|
-
torch_bessel/ops.py,sha256=P4qo1GN6XExuKfI061dsZMpAPuimn6gYt-PwRmR4_Fk,3584
|
6
|
-
torch_bessel-0.0.5.dist-info/LICENSE,sha256=do0DI6wu4mF3VXnEXXPYZqVEatoRSSamgz9t80wU7_o,1068
|
7
|
-
torch_bessel-0.0.5.dist-info/METADATA,sha256=LX6X8j53vJCYKaXLfjlk3X_RwRYHlkoQVYC7CSJJ1vc,1561
|
8
|
-
torch_bessel-0.0.5.dist-info/WHEEL,sha256=QEo1-fvjBiv4iPfIbG-kA4GNtCNK6HOxclrG5aWVgHI,109
|
9
|
-
torch_bessel-0.0.5.dist-info/top_level.txt,sha256=xmyVjWSQ91kX-v8KCzl6wDwfAmbdZNWsP2EH9b9BccQ,24
|
10
|
-
torch_bessel-0.0.5.dist-info/RECORD,,
|
File without changes
|
File without changes
|