torch-linear-assignment 0.0.3__tar.gz → 0.0.4__tar.gz

{torch_linear_assignment-0.0.3/torch_linear_assignment.egg-info → torch_linear_assignment-0.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: torch-linear-assignment
-Version: 0.0.3
+Version: 0.0.4
 Summary: Batched linear assignment with PyTorch and CUDA.
 Author: Ivan Karpukhin
 Author-email: karpuhini@yandex.ru
@@ -12,6 +12,7 @@ Dynamic: author
 Dynamic: author-email
 Dynamic: description
 Dynamic: description-content-type
+Dynamic: license-file
 Dynamic: requires-dist
 Dynamic: summary
 
@@ -40,6 +41,11 @@ Build and install from Git repository:
 pip install .
 ```
 
+Building in an isolated environment may use a different PyTorch version. To match the current environment and reduce the disk usage, apply the following flag:
+```bash
+pip install --no-build-isolation torch-linear-assignment
+```
+
 When building with CUDA, make sure NVCC has the same CUDA version as PyTorch.
 You can choose CUDA version by
 ```bash

{torch_linear_assignment-0.0.3 → torch_linear_assignment-0.0.4}/README.md

@@ -23,6 +23,11 @@ Build and install from Git repository:
 pip install .
 ```
 
+Building in an isolated environment may use a different PyTorch version. To match the current environment and reduce the disk usage, apply the following flag:
+```bash
+pip install --no-build-isolation torch-linear-assignment
+```
+
 When building with CUDA, make sure NVCC has the same CUDA version as PyTorch.
 You can choose CUDA version by
 ```bash

{torch_linear_assignment-0.0.3 → torch_linear_assignment-0.0.4}/setup.py

@@ -48,7 +48,7 @@ with open("README.md") as fp:
 if __name__ == '__main__':
     setuptools.setup(
         name="torch-linear-assignment",
-        version="0.0.3",
+        version="0.0.4",
         author="Ivan Karpukhin",
         author_email="karpuhini@yandex.ru",
         description="Batched linear assignment with PyTorch and CUDA.",

{torch_linear_assignment-0.0.3 → torch_linear_assignment-0.0.4}/src/torch_linear_assignment_cuda_kernel.cu

@@ -35,17 +35,20 @@ int SMPCores(int device_index)
   case 6: // Pascal
     if ((devProp.minor == 1) || (devProp.minor == 2)) return 128;
     else if (devProp.minor == 0) return 64;
+    break;
   case 7: // Volta and Turing
     if ((devProp.minor == 0) || (devProp.minor == 5)) return 64;
+    break;
   case 8: // Ampere
     if (devProp.minor == 0) return 64;
     else if (devProp.minor == 6) return 128;
     else if (devProp.minor == 9) return 128; // ada lovelace
+    break;
   case 9: // Hopper
     if (devProp.minor == 0) return 128;
-  // Unknown device;
+    break;
   }
-  return 128;
+  return 128; // Unknown device
 }
 
 

{torch_linear_assignment-0.0.3 → torch_linear_assignment-0.0.4/torch_linear_assignment.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: torch-linear-assignment
-Version: 0.0.3
+Version: 0.0.4
 Summary: Batched linear assignment with PyTorch and CUDA.
 Author: Ivan Karpukhin
 Author-email: karpuhini@yandex.ru
@@ -12,6 +12,7 @@ Dynamic: author
 Dynamic: author-email
 Dynamic: description
 Dynamic: description-content-type
+Dynamic: license-file
 Dynamic: requires-dist
 Dynamic: summary
 
@@ -40,6 +41,11 @@ Build and install from Git repository:
 pip install .
 ```
 
+Building in an isolated environment may use a different PyTorch version. To match the current environment and reduce the disk usage, apply the following flag:
+```bash
+pip install --no-build-isolation torch-linear-assignment
+```
+
 When building with CUDA, make sure NVCC has the same CUDA version as PyTorch.
 You can choose CUDA version by
 ```bash