refactor: move libs to lib dir

Signed-off-by: Neelay Shah <neelays@nvidia.com>
Co-authored-by: Ryan McCormick <rmccormick@nvidia.com>

python-wheel/Cargo.toml → lib/bindings/python/Cargo.toml

@@ -14,7 +14,7 @@
 # limitations under the License.
 
 [package]
-name = "triton_distributed_py3"
+name = "triton-distributed-py3"
 version = "0.2.0"
 edition = "2021"
 authors = ["NVIDIA"]
@@ -30,8 +30,8 @@ crate-type = ["cdylib"]
 
 
 [dependencies]
-triton-llm = { path = "../llm/rust/triton-llm" }
-triton-distributed = { path = "../runtime/rust" }
+triton-distributed-llm = { path = "../../llm" }
+triton-distributed-runtime = { path = "../../runtime" }
 
 futures = "0.3"
 once_cell = "1.20.3"

python-wheel/examples/bls/bar.py → lib/bindings/python/examples/bls/bar.py

@@ -16,7 +16,8 @@
 import asyncio
 
 import uvloop
-from triton_distributed_rs import DistributedRuntime, triton_worker
+
+from triton_distributed.runtime import DistributedRuntime, triton_worker
 
 uvloop.install()
 

python-wheel/examples/bls/bls.py → lib/bindings/python/examples/bls/bls.py

@@ -16,7 +16,8 @@
 import asyncio
 
 import uvloop
-from triton_distributed_rs import DistributedRuntime, triton_worker
+
+from triton_distributed.runtime import DistributedRuntime, triton_worker
 
 uvloop.install()
 

python-wheel/examples/bls/foo.py → lib/bindings/python/examples/bls/foo.py

@@ -16,7 +16,8 @@
 import asyncio
 
 import uvloop
-from triton_distributed_rs import DistributedRuntime, triton_worker
+
+from triton_distributed.runtime import DistributedRuntime, triton_worker
 
 uvloop.install()
 

python-wheel/examples/error_handling/client.py → lib/bindings/python/examples/error_handling/client.py

@@ -16,7 +16,8 @@
 import asyncio
 
 import uvloop
-from triton_distributed_rs import DistributedRuntime, triton_worker
+
+from triton_distributed.runtime import DistributedRuntime, triton_worker
 
 
 @triton_worker()

python-wheel/examples/hello_world/run.py → lib/bindings/python/examples/error_handling/run.py

@@ -20,7 +20,8 @@ import string
 import uvloop
 from client import init as client_init
 from server import init as server_init
-from triton_distributed_rs import DistributedRuntime, triton_worker
+
+from triton_distributed.runtime import DistributedRuntime, triton_worker
 
 
 def random_string(length=10):

python-wheel/examples/error_handling/server.py → lib/bindings/python/examples/error_handling/server.py

@@ -16,7 +16,8 @@
 import asyncio
 
 import uvloop
-from triton_distributed_rs import DistributedRuntime, triton_worker
+
+from triton_distributed.runtime import DistributedRuntime, triton_worker
 
 
 class RequestHandler:

python-wheel/examples/hello_world/client.py → lib/bindings/python/examples/hello_world/client.py

@@ -16,7 +16,8 @@
 import asyncio
 
 import uvloop
-from triton_distributed_rs import DistributedRuntime, triton_worker
+
+from triton_distributed.runtime import DistributedRuntime, triton_worker
 
 
 @triton_worker()

python-wheel/examples/error_handling/run.py → lib/bindings/python/examples/hello_world/run.py

@@ -20,7 +20,8 @@ import string
 import uvloop
 from client import init as client_init
 from server import init as server_init
-from triton_distributed_rs import DistributedRuntime, triton_worker
+
+from triton_distributed.runtime import DistributedRuntime, triton_worker
 
 
 def random_string(length=10):

python-wheel/examples/hello_world/server.py → lib/bindings/python/examples/hello_world/server.py

@@ -16,7 +16,8 @@
 import asyncio
 
 import uvloop
-from triton_distributed_rs import DistributedRuntime, triton_worker
+
+from triton_distributed.runtime import DistributedRuntime, triton_worker
 
 
 class RequestHandler:

python-wheel/examples/pipeline/backend.py → lib/bindings/python/examples/pipeline/backend.py

@@ -16,7 +16,8 @@
 import asyncio
 
 import uvloop
-from triton_distributed_rs import DistributedRuntime, triton_worker
+
+from triton_distributed.runtime import DistributedRuntime, triton_worker
 
 uvloop.install()
 

python-wheel/examples/pipeline/frontend.py → lib/bindings/python/examples/pipeline/frontend.py

@@ -16,7 +16,8 @@
 import asyncio
 
 import uvloop
-from triton_distributed_rs import DistributedRuntime, triton_worker
+
+from triton_distributed.runtime import DistributedRuntime, triton_worker
 
 uvloop.install()
 

python-wheel/examples/pipeline/middle.py → lib/bindings/python/examples/pipeline/middle.py

@@ -16,7 +16,8 @@
 import asyncio
 
 import uvloop
-from triton_distributed_rs import DistributedRuntime, triton_worker
+
+from triton_distributed.runtime import DistributedRuntime, triton_worker
 
 uvloop.install()
 

python-wheel/examples/pipeline/pipeline.py → lib/bindings/python/examples/pipeline/pipeline.py

@@ -16,7 +16,8 @@
 import asyncio
 
 import uvloop
-from triton_distributed_rs import DistributedRuntime, triton_worker
+
+from triton_distributed.runtime import DistributedRuntime, triton_worker
 
 uvloop.install()