rustat-python-api 0.6.2__tar.gz → 0.6.3__tar.gz
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.
- {rustat-python-api-0.6.2/rustat_python_api.egg-info → rustat-python-api-0.6.3}/PKG-INFO +1 -1
- {rustat-python-api-0.6.2 → rustat-python-api-0.6.3}/rustat_python_api/pitch_control.py +2 -2
- {rustat-python-api-0.6.2 → rustat-python-api-0.6.3/rustat_python_api.egg-info}/PKG-INFO +1 -1
- {rustat-python-api-0.6.2 → rustat-python-api-0.6.3}/setup.py +1 -1
- {rustat-python-api-0.6.2 → rustat-python-api-0.6.3}/LICENSE +0 -0
- {rustat-python-api-0.6.2 → rustat-python-api-0.6.3}/README.md +0 -0
- {rustat-python-api-0.6.2 → rustat-python-api-0.6.3}/pyproject.toml +0 -0
- {rustat-python-api-0.6.2 → rustat-python-api-0.6.3}/rustat_python_api/__init__.py +0 -0
- {rustat-python-api-0.6.2 → rustat-python-api-0.6.3}/rustat_python_api/config.py +0 -0
- {rustat-python-api-0.6.2 → rustat-python-api-0.6.3}/rustat_python_api/kernels/__init__.py +0 -0
- {rustat-python-api-0.6.2 → rustat-python-api-0.6.3}/rustat_python_api/kernels/maha.py +0 -0
- {rustat-python-api-0.6.2 → rustat-python-api-0.6.3}/rustat_python_api/models_api.py +0 -0
- {rustat-python-api-0.6.2 → rustat-python-api-0.6.3}/rustat_python_api/parser.py +0 -0
- {rustat-python-api-0.6.2 → rustat-python-api-0.6.3}/rustat_python_api/processing.py +0 -0
- {rustat-python-api-0.6.2 → rustat-python-api-0.6.3}/rustat_python_api/urls.py +0 -0
- {rustat-python-api-0.6.2 → rustat-python-api-0.6.3}/rustat_python_api.egg-info/SOURCES.txt +0 -0
- {rustat-python-api-0.6.2 → rustat-python-api-0.6.3}/rustat_python_api.egg-info/dependency_links.txt +0 -0
- {rustat-python-api-0.6.2 → rustat-python-api-0.6.3}/rustat_python_api.egg-info/requires.txt +0 -0
- {rustat-python-api-0.6.2 → rustat-python-api-0.6.3}/rustat_python_api.egg-info/top_level.txt +0 -0
- {rustat-python-api-0.6.2 → rustat-python-api-0.6.3}/setup.cfg +0 -0
|
@@ -322,7 +322,7 @@ class PitchControl:
|
|
|
322
322
|
diff = locs.view(1, -1, 2) # (1,N,2)
|
|
323
323
|
diff = diff - mu.unsqueeze(1) # (P,N,2)
|
|
324
324
|
|
|
325
|
-
if
|
|
325
|
+
if "cuda" in device:
|
|
326
326
|
out = triton_influence(
|
|
327
327
|
mu.unsqueeze(0), Sigma_inv.unsqueeze(0),
|
|
328
328
|
locs, BLOCK_N=64
|
|
@@ -391,7 +391,7 @@ class PitchControl:
|
|
|
391
391
|
diff = locs.view(1, 1, -1, 2) # (1,1,N,2)
|
|
392
392
|
diff = diff - mu.unsqueeze(2) # (F,P,N,2)
|
|
393
393
|
|
|
394
|
-
if
|
|
394
|
+
if "cuda" in device:
|
|
395
395
|
out = triton_influence(mu, Sigma_inv, locs, BLOCK_N=64) # (F,N)
|
|
396
396
|
|
|
397
397
|
return out
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
{rustat-python-api-0.6.2 → rustat-python-api-0.6.3}/rustat_python_api.egg-info/dependency_links.txt
RENAMED
|
File without changes
|
|
File without changes
|
{rustat-python-api-0.6.2 → rustat-python-api-0.6.3}/rustat_python_api.egg-info/top_level.txt
RENAMED
|
File without changes
|
|
File without changes
|