mlx-cluster 0.0.1__tar.gz → 0.0.2__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.
Files changed (26) hide show
  1. {mlx_cluster-0.0.1/mlx_cluster.egg-info → mlx_cluster-0.0.2}/PKG-INFO +10 -4
  2. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/README.md +9 -3
  3. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/mlx_cluster/_ext.cpython-311-darwin.so +0 -0
  4. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/mlx_cluster/libmlx_cluster.dylib +0 -0
  5. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2/mlx_cluster.egg-info}/PKG-INFO +10 -4
  6. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/pyproject.toml +3 -2
  7. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/setup.py +1 -1
  8. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/CMakeLists.txt +0 -0
  9. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/LICENSE +0 -0
  10. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/MANIFEST.in +0 -0
  11. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/bindings.cpp +0 -0
  12. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/mlx_cluster/__init__.py +0 -0
  13. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/mlx_cluster/mlx_cluster.metallib +0 -0
  14. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/mlx_cluster.egg-info/SOURCES.txt +0 -0
  15. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/mlx_cluster.egg-info/dependency_links.txt +0 -0
  16. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/mlx_cluster.egg-info/not-zip-safe +0 -0
  17. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/mlx_cluster.egg-info/requires.txt +0 -0
  18. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/mlx_cluster.egg-info/top_level.txt +0 -0
  19. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/random_walks/BiasedRandomWalk.cpp +0 -0
  20. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/random_walks/BiasedRandomWalk.h +0 -0
  21. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/random_walks/RandomWalk.cpp +0 -0
  22. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/random_walks/RandomWalk.h +0 -0
  23. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/random_walks/random_walk.metal +0 -0
  24. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/setup.cfg +0 -0
  25. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/tests/test_random_walk.py +0 -0
  26. {mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/tests/test_rejection_sampling.py +0 -0
{mlx_cluster-0.0.1/mlx_cluster.egg-info → mlx_cluster-0.0.2}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: mlx_cluster
3
- Version: 0.0.1
3
+ Version: 0.0.2
4
4
  Summary: C++ and Metal extensions for MLX CTC Loss
5
5
  Author-email: Vinay Pandya <vinayharshadpandya27@gmail.com>
6
6
  Project-URL: Homepage, https://github.com/vinayhpandya/mlx_cluster
@@ -20,7 +20,7 @@ Requires-Dist: pytest; extra == "test"
20
20
 
21
21
  # mlx_cluster
22
22
 
23
- A C++ extension for generating ramdom walks for Homogeneous graphs using mlx
23
+ A C++ extension for generating random walks for Homogeneous graphs using mlx
24
24
 
25
25
  ## Installation
26
26
 
@@ -37,7 +37,13 @@ After cloning the repository install library using
37
37
  python setup.py build_ext -j8 --inplace
38
38
  ```
39
39
 
40
- for testing purposes you need to have `mlx-graphs` installed
40
+ You can also just install the library via pip
41
+
42
+ ```bash
43
+ pip install mlx_cluster
44
+ ```
45
+
46
+ for testing purposes you need to have `mlx-graphs` and `torch_geometric` installed
41
47
 
42
48
  ## Usage
43
49
 
@@ -51,7 +57,7 @@ from mlx_graphs_extension import random_walk
51
57
  cora_dataset = PlanetoidDataset(name="cora", base_dir="~")
52
58
  start = mx.arange(0, 1000)
53
59
  start_time = time.time()
54
- edge_index = cora_dataset.graphs[0].edge_index
60
+ edge_index = cora_dataset.graphs[0].edge_index.astype(mx.int64)
55
61
  num_nodes = cora_dataset.graphs[0].num_nodes
56
62
  sorted_edge_index = sort_edge_index(edge_index=edge_index)
57
63
  row_mlx = sorted_edge_index[0][0]
{mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/README.md RENAMED
@@ -1,6 +1,6 @@
1
1
  # mlx_cluster
2
2
 
3
- A C++ extension for generating ramdom walks for Homogeneous graphs using mlx
3
+ A C++ extension for generating random walks for Homogeneous graphs using mlx
4
4
 
5
5
  ## Installation
6
6
 
@@ -17,7 +17,13 @@ After cloning the repository install library using
17
17
  python setup.py build_ext -j8 --inplace
18
18
  ```
19
19
 
20
- for testing purposes you need to have `mlx-graphs` installed
20
+ You can also just install the library via pip
21
+
22
+ ```bash
23
+ pip install mlx_cluster
24
+ ```
25
+
26
+ for testing purposes you need to have `mlx-graphs` and `torch_geometric` installed
21
27
 
22
28
  ## Usage
23
29
 
@@ -31,7 +37,7 @@ from mlx_graphs_extension import random_walk
31
37
  cora_dataset = PlanetoidDataset(name="cora", base_dir="~")
32
38
  start = mx.arange(0, 1000)
33
39
  start_time = time.time()
34
- edge_index = cora_dataset.graphs[0].edge_index
40
+ edge_index = cora_dataset.graphs[0].edge_index.astype(mx.int64)
35
41
  num_nodes = cora_dataset.graphs[0].num_nodes
36
42
  sorted_edge_index = sort_edge_index(edge_index=edge_index)
37
43
  row_mlx = sorted_edge_index[0][0]
{mlx_cluster-0.0.1 → mlx_cluster-0.0.2/mlx_cluster.egg-info}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: mlx_cluster
3
- Version: 0.0.1
3
+ Version: 0.0.2
4
4
  Summary: C++ and Metal extensions for MLX CTC Loss
5
5
  Author-email: Vinay Pandya <vinayharshadpandya27@gmail.com>
6
6
  Project-URL: Homepage, https://github.com/vinayhpandya/mlx_cluster
@@ -20,7 +20,7 @@ Requires-Dist: pytest; extra == "test"
20
20
 
21
21
  # mlx_cluster
22
22
 
23
- A C++ extension for generating ramdom walks for Homogeneous graphs using mlx
23
+ A C++ extension for generating random walks for Homogeneous graphs using mlx
24
24
 
25
25
  ## Installation
26
26
 
@@ -37,7 +37,13 @@ After cloning the repository install library using
37
37
  python setup.py build_ext -j8 --inplace
38
38
  ```
39
39
 
40
- for testing purposes you need to have `mlx-graphs` installed
40
+ You can also just install the library via pip
41
+
42
+ ```bash
43
+ pip install mlx_cluster
44
+ ```
45
+
46
+ for testing purposes you need to have `mlx-graphs` and `torch_geometric` installed
41
47
 
42
48
  ## Usage
43
49
 
@@ -51,7 +57,7 @@ from mlx_graphs_extension import random_walk
51
57
  cora_dataset = PlanetoidDataset(name="cora", base_dir="~")
52
58
  start = mx.arange(0, 1000)
53
59
  start_time = time.time()
54
- edge_index = cora_dataset.graphs[0].edge_index
60
+ edge_index = cora_dataset.graphs[0].edge_index.astype(mx.int64)
55
61
  num_nodes = cora_dataset.graphs[0].num_nodes
56
62
  sorted_edge_index = sort_edge_index(edge_index=edge_index)
57
63
  row_mlx = sorted_edge_index[0][0]
{mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "mlx_cluster"
3
- version = "0.0.1"
3
+ version = "0.0.2"
4
4
  authors = [
5
5
  { name = "Vinay Pandya", email = "vinayharshadpandya27@gmail.com" },
6
6
  ]
@@ -27,7 +27,8 @@ Issues = "https://github.com/vinayhpandya/mlx_cluster/Issues"
27
27
  requires = [
28
28
  "setuptools>=42",
29
29
  "cmake>=3.24",
30
- "mlx>=0.9.0",
30
+ "mlx==0.15.*",
31
31
  "nanobind@git+https://github.com/wjakob/nanobind.git@2f04eac452a6d9142dedb957701bdb20125561e4",
32
+ "fmt"
32
33
  ]
33
34
  build-backend = "setuptools.build_meta"
{mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/setup.py RENAMED
@@ -4,7 +4,7 @@ from mlx import extension
4
4
  if __name__ == "__main__":
5
5
  setup(
6
6
  name="mlx_cluster",
7
- version="0.0.1",
7
+ version="0.0.2",
8
8
  description="Sample C++ and Metal extensions for MLX primitives.",
9
9
  ext_modules=[extension.CMakeExtension("mlx_cluster._ext")],
10
10
  cmdclass={"build_ext": extension.CMakeBuild},
{mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/CMakeLists.txt RENAMED
File without changes
{mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/LICENSE RENAMED
File without changes
{mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/MANIFEST.in RENAMED
File without changes
{mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/bindings.cpp RENAMED
File without changes
{mlx_cluster-0.0.1 → mlx_cluster-0.0.2}/setup.cfg RENAMED
File without changes