diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml index b7fb97b95..3ebe827b9 100644 --- a/.github/workflows/publish.yml +++ b/.github/workflows/publish.yml @@ -7,140 +7,57 @@ on: workflow_dispatch: inputs: dry-run-only: - description: "Run xtask publish in dry-run mode (no publish)" + description: "Run crate publish jobs in dry-run mode (no publish)" type: boolean required: false default: false jobs: - publish-cubek-random: + publish-t4a-cubek-random: uses: tracel-ai/github-actions/.github/workflows/publish-crate.yml@v8 with: - crate: cubek-random + crate: t4a-cubek-random dry-run-only: ${{ github.event_name == 'workflow_dispatch' && inputs.dry-run-only || false }} secrets: CRATES_IO_API_TOKEN: ${{ secrets.CRATES_IO_API_TOKEN }} - publish-cubek-test-utils: - needs: - - publish-cubek-random - - publish-cubek-quant - uses: tracel-ai/github-actions/.github/workflows/publish-crate.yml@v8 - with: - crate: cubek-test-utils - dry-run-only: ${{ github.event_name == 'workflow_dispatch' && inputs.dry-run-only || false }} - secrets: - CRATES_IO_API_TOKEN: ${{ secrets.CRATES_IO_API_TOKEN }} - - publish-cubek-quant: - uses: tracel-ai/github-actions/.github/workflows/publish-crate.yml@v8 - with: - crate: cubek-quant - dry-run-only: ${{ github.event_name == 'workflow_dispatch' && inputs.dry-run-only || false }} - secrets: - CRATES_IO_API_TOKEN: ${{ secrets.CRATES_IO_API_TOKEN }} - - publish-cubek-reduce: - needs: - - publish-cubek-random - - publish-cubek-test-utils - - publish-cubek-std - uses: tracel-ai/github-actions/.github/workflows/publish-crate.yml@v8 - with: - crate: cubek-reduce - dry-run-only: ${{ github.event_name == 'workflow_dispatch' && inputs.dry-run-only || false }} - secrets: - CRATES_IO_API_TOKEN: ${{ secrets.CRATES_IO_API_TOKEN }} - - publish-cubek-matmul: - needs: - - publish-cubek-random - - publish-cubek-reduce - - publish-cubek-quant - - publish-cubek-test-utils - - publish-cubek-std + publish-t4a-cubek-quant: uses: tracel-ai/github-actions/.github/workflows/publish-crate.yml@v8 with: - crate: cubek-matmul + crate: t4a-cubek-quant dry-run-only: ${{ github.event_name == 'workflow_dispatch' && inputs.dry-run-only || false }} secrets: CRATES_IO_API_TOKEN: ${{ secrets.CRATES_IO_API_TOKEN }} - publish-cubek-convolution: - needs: - - publish-cubek-random - - publish-cubek-reduce - - publish-cubek-matmul - - publish-cubek-test-utils - - publish-cubek-std - uses: tracel-ai/github-actions/.github/workflows/publish-crate.yml@v8 - with: - crate: cubek-convolution - dry-run-only: ${{ github.event_name == 'workflow_dispatch' && inputs.dry-run-only || false }} - secrets: - CRATES_IO_API_TOKEN: ${{ secrets.CRATES_IO_API_TOKEN }} - - publish-cubek-attention: - needs: - - publish-cubek-random - - publish-cubek-reduce - - publish-cubek-matmul - - publish-cubek-test-utils - - publish-cubek-std - uses: tracel-ai/github-actions/.github/workflows/publish-crate.yml@v8 - with: - crate: cubek-attention - dry-run-only: ${{ github.event_name == 'workflow_dispatch' && inputs.dry-run-only || false }} - secrets: - CRATES_IO_API_TOKEN: ${{ secrets.CRATES_IO_API_TOKEN }} - - publish-cubek-fft: - needs: - - publish-cubek-test-utils - - publish-cubek-std - uses: tracel-ai/github-actions/.github/workflows/publish-crate.yml@v8 - with: - crate: cubek-fft - dry-run-only: ${{ github.event_name == 'workflow_dispatch' && inputs.dry-run-only || false }} - secrets: - CRATES_IO_API_TOKEN: ${{ secrets.CRATES_IO_API_TOKEN }} - - publish-cubek-interpolate: + publish-t4a-cubek-test-utils: needs: - - publish-cubek-test-utils + - publish-t4a-cubek-random + - publish-t4a-cubek-quant uses: tracel-ai/github-actions/.github/workflows/publish-crate.yml@v8 with: - crate: cubek-interpolate + crate: t4a-cubek-test-utils dry-run-only: ${{ github.event_name == 'workflow_dispatch' && inputs.dry-run-only || false }} secrets: CRATES_IO_API_TOKEN: ${{ secrets.CRATES_IO_API_TOKEN }} - - publish-cubek-std: + publish-t4a-cubek-std: needs: - - publish-cubek-test-utils + - publish-t4a-cubek-test-utils uses: tracel-ai/github-actions/.github/workflows/publish-crate.yml@v8 with: - crate: cubek-std + crate: t4a-cubek-std dry-run-only: ${{ github.event_name == 'workflow_dispatch' && inputs.dry-run-only || false }} secrets: CRATES_IO_API_TOKEN: ${{ secrets.CRATES_IO_API_TOKEN }} - publish-cubek: + publish-t4a-cubek-matmul: needs: - - publish-cubek-random - - publish-cubek-reduce - - publish-cubek-quant - - publish-cubek-matmul - - publish-cubek-attention - - publish-cubek-convolution - - publish-cubek-fft - - publish-cubek-interpolate - - publish-cubek-test-utils - - publish-cubek-std + - publish-t4a-cubek-quant + - publish-t4a-cubek-std + - publish-t4a-cubek-test-utils uses: tracel-ai/github-actions/.github/workflows/publish-crate.yml@v8 with: - crate: cubek + crate: t4a-cubek-matmul dry-run-only: ${{ github.event_name == 'workflow_dispatch' && inputs.dry-run-only || false }} secrets: CRATES_IO_API_TOKEN: ${{ secrets.CRATES_IO_API_TOKEN }} diff --git a/Cargo.toml b/Cargo.toml index a3f00bd70..f08ec35fa 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -15,17 +15,12 @@ rust-version = "1.92" version = "0.2.0" [workspace.dependencies] -### For local development. ### -# cubecl = { path = "../cubecl/crates/cubecl", default-features = false } -# cubecl-common = { path = "../cubecl/crates/cubecl-common", default-features = false } - -### For the main cubek branch. ### -# cubecl = { git = "https://github.com/tracel-ai/cubecl", rev = "7cf203735e095e640a2c03b2400d0faa03196bb4", default-features = false } -# cubecl-common = { git = "https://github.com/tracel-ai/cubecl", rev = "7cf203735e095e640a2c03b2400d0faa03196bb4", default-features = false } - -### For tensor4all fork integration. ### -cubecl = { git = "https://github.com/tensor4all/cubecl.git", rev = "f5e5ec178f9aebca9362b829ffef708f720ff692", default-features = false } -cubecl-common = { git = "https://github.com/tensor4all/cubecl.git", rev = "f5e5ec178f9aebca9362b829ffef708f720ff692", default-features = false } +### tensor4all fork release preparation. ### +# PR/CI builds use the tensor4all CubeCL commit that adds the `t4a-cubecl*` +# package manifests. Published packages use the registry packages at the same +# exact versions. +cubecl = { package = "t4a-cubecl", git = "https://github.com/tensor4all/cubecl.git", rev = "68365e47322123359d0a07d741ba9771a0bf6ffe", version = "=0.10.0", default-features = false } +cubecl-common = { package = "t4a-cubecl-common", git = "https://github.com/tensor4all/cubecl.git", rev = "68365e47322123359d0a07d741ba9771a0bf6ffe", version = "=0.10.0", default-features = false } derive-new = { version = "0.7.0", default-features = false } log = { default-features = false, version = "0.4.22" } diff --git a/README.md b/README.md index 1b7e09add..19b4d2a7e 100644 --- a/README.md +++ b/README.md @@ -1,3 +1,9 @@ +> **tensor4all temporary fork.** The `t4a-cubek*` crates published from this +> repository are temporary tensor4all fork packages used by `tenferro-rs` while +> required CubeK/CubeCL patches are being upstreamed. They are not a replacement +> for upstream CubeK, and the intent is to return to upstream package names once +> the needed patches have landed. +
diff --git a/benchmarks/Cargo.toml b/benchmarks/Cargo.toml index 0bccda858..538e3b1e0 100644 --- a/benchmarks/Cargo.toml +++ b/benchmarks/Cargo.toml @@ -9,8 +9,9 @@ edition.workspace = true keywords = [] license.workspace = true name = "benchmarks" +publish = false readme.workspace = true -repository = "https://github.com/tracel-ai/cubek" +repository = "https://github.com/tensor4all/cubek" version.workspace = true [features] @@ -46,7 +47,7 @@ cubek = { path = "../crates/cubek", version = "0.2.0", default-features = false, cubek-interpolate = { path = "../crates/cubek-interpolate", version = "0.2.0",features = [ "cpu-reference", ], optional = true } -cubek-matmul = { path = "../crates/cubek-matmul", version = "0.2.0", features = [ +cubek-matmul = { package = "t4a-cubek-matmul", path = "../crates/cubek-matmul", version = "0.2.0", features = [ "cpu-reference", ], optional = true } cubek-attention = { path = "../crates/cubek-attention", version = "0.2.0", features = [ @@ -61,7 +62,7 @@ cubek-fft = { path = "../crates/cubek-fft", version = "0.2.0", features = [ cubek-reduce = { path = "../crates/cubek-reduce", version = "0.2.0", features = [ "cpu-reference", ], optional = true } -cubek-test-utils = { path = "../crates/cubek-test-utils", version = "0.2.0" } +cubek-test-utils = { package = "t4a-cubek-test-utils", path = "../crates/cubek-test-utils", version = "0.2.0" } half = { workspace = true } [[bench]] diff --git a/crates/cubek-attention/Cargo.toml b/crates/cubek-attention/Cargo.toml index 2af638246..de76fef35 100644 --- a/crates/cubek-attention/Cargo.toml +++ b/crates/cubek-attention/Cargo.toml @@ -6,8 +6,9 @@ edition.workspace = true keywords = [] license.workspace = true name = "cubek-attention" +publish = false readme.workspace = true -repository = "https://github.com/tracel-ai/cubek/tree/main/crates/cubek-attention" +repository = "https://github.com/tensor4all/cubek/tree/main/crates/cubek-attention" version.workspace = true [features] @@ -23,9 +24,9 @@ cpu-reference = ["dep:cubek-test-utils"] [dependencies] cubecl = { workspace = true, features = ["stdlib"] } cubecl-common = { workspace = true } -cubek-matmul = { path = "../cubek-matmul", version = "0.2.0", default-features = false } -cubek-std = { path = "../cubek-std", version = "0.2.0", default-features = false } -cubek-test-utils = { path = "../cubek-test-utils", version = "0.2.0", default-features = false, optional = true } +cubek-matmul = { package = "t4a-cubek-matmul", path = "../cubek-matmul", version = "0.2.0", default-features = false } +cubek-std = { package = "t4a-cubek-std", path = "../cubek-std", version = "0.2.0", default-features = false } +cubek-test-utils = { package = "t4a-cubek-test-utils", path = "../cubek-test-utils", version = "0.2.0", default-features = false, optional = true } bytemuck = { workspace = true } half = { workspace = true, features = ["bytemuck"] } @@ -34,6 +35,6 @@ serde = { workspace = true } [dev-dependencies] cubecl = { workspace = true, features = ["test-runtime"] } cubek-attention = { path = ".", features = ["cpu-reference"] } -cubek-test-utils = { path = "../cubek-test-utils", version = "0.2.0", default-features = false } +cubek-test-utils = { package = "t4a-cubek-test-utils", path = "../cubek-test-utils", version = "0.2.0", default-features = false } pretty_assertions = { workspace = true } trybuild = "1" diff --git a/crates/cubek-convolution/Cargo.toml b/crates/cubek-convolution/Cargo.toml index 78b2c19c5..b1e45674b 100644 --- a/crates/cubek-convolution/Cargo.toml +++ b/crates/cubek-convolution/Cargo.toml @@ -6,8 +6,9 @@ edition.workspace = true keywords = [] license.workspace = true name = "cubek-convolution" +publish = false readme.workspace = true -repository = "https://github.com/tracel-ai/cubek/tree/main/crates/cubek-convolution" +repository = "https://github.com/tensor4all/cubek/tree/main/crates/cubek-convolution" version.workspace = true [features] @@ -23,9 +24,9 @@ cpu-reference = ["dep:cubek-test-utils", "cubek-matmul/cpu-reference"] [dependencies] cubecl = { workspace = true, features = ["stdlib"] } cubecl-common = { workspace = true } -cubek-matmul = { path = "../cubek-matmul", default-features = false, version = "0.2.0" } -cubek-std = { path = "../cubek-std", default-features = false, version = "0.2.0" } -cubek-test-utils = { path = "../cubek-test-utils", version = "0.2.0", default-features = false, optional = true } +cubek-matmul = { package = "t4a-cubek-matmul", path = "../cubek-matmul", default-features = false, version = "0.2.0" } +cubek-std = { package = "t4a-cubek-std", path = "../cubek-std", default-features = false, version = "0.2.0" } +cubek-test-utils = { package = "t4a-cubek-test-utils", path = "../cubek-test-utils", version = "0.2.0", default-features = false, optional = true } derive-new = { workspace = true } bytemuck = { workspace = true } @@ -36,6 +37,6 @@ serde = { workspace = true } [dev-dependencies] cubecl = { workspace = true, features = ["test-runtime"] } cubek-convolution = { path = ".", features = ["cpu-reference"] } -cubek-test-utils = { path = "../cubek-test-utils", version = "0.2.0", default-features = false } +cubek-test-utils = { package = "t4a-cubek-test-utils", path = "../cubek-test-utils", version = "0.2.0", default-features = false } pretty_assertions = { workspace = true } trybuild = "1" diff --git a/crates/cubek-fft/Cargo.toml b/crates/cubek-fft/Cargo.toml index 3fb1fc566..b21211892 100644 --- a/crates/cubek-fft/Cargo.toml +++ b/crates/cubek-fft/Cargo.toml @@ -6,8 +6,9 @@ edition.workspace = true keywords = [] license.workspace = true name = "cubek-fft" +publish = false readme.workspace = true -repository = "https://github.com/tracel-ai/cubek/tree/main/crates/cubek-fft" +repository = "https://github.com/tensor4all/cubek/tree/main/crates/cubek-fft" version.workspace = true [features] @@ -22,7 +23,7 @@ cpu-reference = ["dep:num-complex", "dep:cubek-test-utils"] [dependencies] cubecl = { workspace = true } -cubek-test-utils = { path = "./../cubek-test-utils/", version = "0.2.0", default-features = false, optional = true } +cubek-test-utils = { package = "t4a-cubek-test-utils", path = "./../cubek-test-utils/", version = "0.2.0", default-features = false, optional = true } num-complex = { workspace = true, optional = true } [dev-dependencies] @@ -30,4 +31,4 @@ num-complex = { workspace = true } cubecl = { workspace = true, features = ["test-runtime"] } cubecl-common = { workspace = true } cubek-fft = { path = ".", features = ["cpu-reference"] } -cubek-test-utils = { path = "./../cubek-test-utils/", version = "0.2.0", default-features = false } +cubek-test-utils = { package = "t4a-cubek-test-utils", path = "./../cubek-test-utils/", version = "0.2.0", default-features = false } diff --git a/crates/cubek-interpolate/Cargo.toml b/crates/cubek-interpolate/Cargo.toml index 9e7b5a31f..91ab9071b 100644 --- a/crates/cubek-interpolate/Cargo.toml +++ b/crates/cubek-interpolate/Cargo.toml @@ -6,8 +6,9 @@ edition.workspace = true keywords = [] license.workspace = true name = "cubek-interpolate" +publish = false readme.workspace = true -repository = "https://github.com/tracel-ai/cubek/tree/main/crates/cubek-interpolate" +repository = "https://github.com/tensor4all/cubek/tree/main/crates/cubek-interpolate" version.workspace = true [features] @@ -24,9 +25,9 @@ cpu-reference = ["dep:cubek-test-utils"] cubecl = { workspace = true } cubecl-common = { workspace = true } thiserror = { workspace = true } -cubek-test-utils = { path = "./../cubek-test-utils/", version = "0.2.0", default-features = false, optional = true } +cubek-test-utils = { package = "t4a-cubek-test-utils", path = "./../cubek-test-utils/", version = "0.2.0", default-features = false, optional = true } [dev-dependencies] cubecl = { workspace = true, features = ["test-runtime"] } cubek-interpolate = { path = ".", features = ["cpu-reference"] } -cubek-test-utils = { path = "../cubek-test-utils", version = "0.2.0", default-features = false } +cubek-test-utils = { package = "t4a-cubek-test-utils", path = "../cubek-test-utils", version = "0.2.0", default-features = false } diff --git a/crates/cubek-matmul/Cargo.toml b/crates/cubek-matmul/Cargo.toml index e43c17a50..ac4a0840f 100644 --- a/crates/cubek-matmul/Cargo.toml +++ b/crates/cubek-matmul/Cargo.toml @@ -1,15 +1,18 @@ [package] authors = ["louisfd "] categories = ["science", "mathematics", "algorithms"] -description = "CubeK: Matrix Multiplication Kernels" +description = "Temporary tensor4all fork of CubeK: Matrix Multiplication Kernels" edition.workspace = true keywords = [] license.workspace = true -name = "cubek-matmul" -readme.workspace = true -repository = "https://github.com/tracel-ai/cubek/tree/main/crates/cubek-matmul" +name = "t4a-cubek-matmul" +readme = "README.md" +repository = "https://github.com/tensor4all/cubek/tree/main/crates/cubek-matmul" version.workspace = true +[lib] +name = "cubek_matmul" + [features] default = ["std", "cubecl/default"] std = ["cubecl/std"] @@ -21,8 +24,8 @@ full = ["extended"] cpu-reference = ["dep:cubek-test-utils"] [dependencies] -cubek-std = { path = "../cubek-std", version = "0.2.0", default-features = false } -cubek-test-utils = { path = "../cubek-test-utils", version = "0.2.0", default-features = false, optional = true } +cubek-std = { package = "t4a-cubek-std", path = "../cubek-std", version = "0.2.0", default-features = false } +cubek-test-utils = { package = "t4a-cubek-test-utils", path = "../cubek-test-utils", version = "0.2.0", default-features = false, optional = true } bytemuck = { workspace = true } cubecl = { workspace = true, features = ["stdlib"] } @@ -33,9 +36,9 @@ serde = { workspace = true } [dev-dependencies] cubecl = { workspace = true, features = ["test-runtime"] } -cubek-matmul = { path = ".", features = ["cpu-reference"] } -cubek-test-utils = { path = "../cubek-test-utils", version = "0.2.0", default-features = false } -cubek-quant = { path = "../cubek-quant", version = "0.2.0", features = [ +cubek-matmul = { package = "t4a-cubek-matmul", path = ".", features = ["cpu-reference"] } +cubek-test-utils = { package = "t4a-cubek-test-utils", path = "../cubek-test-utils", version = "0.2.0", default-features = false } +cubek-quant = { package = "t4a-cubek-quant", path = "../cubek-quant", version = "0.2.0", features = [ "kernels", ] } pretty_assertions = { workspace = true } diff --git a/crates/cubek-matmul/README.md b/crates/cubek-matmul/README.md index 80dd136f5..c9708bb6b 100644 --- a/crates/cubek-matmul/README.md +++ b/crates/cubek-matmul/README.md @@ -1 +1,3 @@ +> **tensor4all temporary fork.** This `t4a-cubek*` package is a temporary tensor4all fork used by `tenferro-rs` while required patches are being upstreamed to the official CubeK/CubeCL projects. It is not a replacement for upstream CubeK. + # CubeCL Matrix Multiplication Kernels Engine diff --git a/crates/cubek-quant/Cargo.toml b/crates/cubek-quant/Cargo.toml index 4795f9092..c0d24a940 100644 --- a/crates/cubek-quant/Cargo.toml +++ b/crates/cubek-quant/Cargo.toml @@ -1,15 +1,18 @@ [package] authors = ["nathanielsimard "] categories = ["science", "mathematics", "algorithms"] -description = "CubeK: Quantization Library" +description = "Temporary tensor4all fork of CubeK: Quantization Library" edition.workspace = true keywords = [] license.workspace = true -name = "cubek-quant" +name = "t4a-cubek-quant" readme.workspace = true -repository = "https://github.com/tracel-ai/cubek/tree/main/crates/cubek-quant" +repository = "https://github.com/tensor4all/cubek/tree/main/crates/cubek-quant" version.workspace = true +[lib] +name = "cubek_quant" + [features] default = ["kernels"] kernels = ["std"] diff --git a/crates/cubek-random/Cargo.toml b/crates/cubek-random/Cargo.toml index 961dedd69..631e0ca79 100644 --- a/crates/cubek-random/Cargo.toml +++ b/crates/cubek-random/Cargo.toml @@ -1,15 +1,18 @@ [package] authors = ["nathanielsimard "] categories = ["science", "mathematics", "algorithms"] -description = "CubeK: Random Number Generation" +description = "Temporary tensor4all fork of CubeK: Random Number Generation" edition.workspace = true keywords = [] license.workspace = true -name = "cubek-random" +name = "t4a-cubek-random" readme.workspace = true -repository = "https://github.com/tracel-ai/cubek/tree/main/crates/cubek-random" +repository = "https://github.com/tensor4all/cubek/tree/main/crates/cubek-random" version.workspace = true +[lib] +name = "cubek_random" + [features] default = ["std", "cubecl/default"] std = ["cubecl/std"] diff --git a/crates/cubek-reduce/Cargo.toml b/crates/cubek-reduce/Cargo.toml index f3894218b..b61534feb 100644 --- a/crates/cubek-reduce/Cargo.toml +++ b/crates/cubek-reduce/Cargo.toml @@ -10,8 +10,9 @@ edition.workspace = true keywords = [] license.workspace = true name = "cubek-reduce" +publish = false readme.workspace = true -repository = "https://github.com/tracel-ai/cubek/tree/main/crates/cubek-reduce" +repository = "https://github.com/tensor4all/cubek/tree/main/crates/cubek-reduce" version.workspace = true [features] @@ -26,8 +27,8 @@ cpu-reference = ["dep:cubek-test-utils"] [dependencies] cubecl = { workspace = true } -cubek-std = { path = "../cubek-std", version = "0.2.0", default-features = false } -cubek-test-utils = { path = "./../cubek-test-utils/", version = "0.2.0", default-features = false, optional = true } +cubek-std = { package = "t4a-cubek-std", path = "../cubek-std", version = "0.2.0", default-features = false } +cubek-test-utils = { package = "t4a-cubek-test-utils", path = "./../cubek-test-utils/", version = "0.2.0", default-features = false, optional = true } num-traits = "0.2.19" pretty_assertions = { workspace = true, optional = true } @@ -39,4 +40,4 @@ thiserror = { workspace = true } cubecl = { workspace = true, features = ["test-runtime"] } cubecl-common = { workspace = true } cubek-reduce = { path = ".", features = ["cpu-reference"] } -cubek-test-utils = { path = "./../cubek-test-utils/", version = "0.2.0" } +cubek-test-utils = { package = "t4a-cubek-test-utils", path = "./../cubek-test-utils/", version = "0.2.0" } diff --git a/crates/cubek-std/Cargo.toml b/crates/cubek-std/Cargo.toml index 42c6c5309..d0efe514c 100644 --- a/crates/cubek-std/Cargo.toml +++ b/crates/cubek-std/Cargo.toml @@ -1,14 +1,17 @@ [package] authors = ["louisfd "] -description = "CubeK: Standard Library" +description = "Temporary tensor4all fork of CubeK: Standard Library" edition.workspace = true license.workspace = true -name = "cubek-std" +name = "t4a-cubek-std" readme.workspace = true rust-version.workspace = true -repository = "https://github.com/tracel-ai/cubek/tree/main/crates/cubek-std" +repository = "https://github.com/tensor4all/cubek/tree/main/crates/cubek-std" version.workspace = true +[lib] +name = "cubek_std" + [features] default = ["cubecl/default"] std = ["cubecl/std"] @@ -23,4 +26,4 @@ cubecl-common = { workspace = true } half = { workspace = true, features = ["bytemuck"] } [dev-dependencies] -cubek-test-utils = { path = "../cubek-test-utils", version = "0.2.0", default-features = false } +cubek-test-utils = { package = "t4a-cubek-test-utils", path = "../cubek-test-utils", version = "0.2.0", default-features = false } diff --git a/crates/cubek-test-utils/Cargo.toml b/crates/cubek-test-utils/Cargo.toml index 836df9c3f..a93bfb336 100644 --- a/crates/cubek-test-utils/Cargo.toml +++ b/crates/cubek-test-utils/Cargo.toml @@ -3,15 +3,18 @@ authors = [ "nathanielsimard ", "louisfd ", ] -description = "CubeK: Test Utils" +description = "Temporary tensor4all fork of CubeK: Test Utils" edition.workspace = true license.workspace = true -name = "cubek-test-utils" -readme.workspace = true +name = "t4a-cubek-test-utils" +readme = "README.md" rust-version.workspace = true -repository = "https://github.com/tracel-ai/cubek/tree/main/crates/cubek-test-utils" +repository = "https://github.com/tensor4all/cubek/tree/main/crates/cubek-test-utils" version.workspace = true +[lib] +name = "cubek_test_utils" + [features] default = ["cubecl/default"] std = ["cubecl/std"] @@ -21,8 +24,8 @@ extended = ["heavy"] full = ["extended"] [dependencies] -cubek-random = { path = "../cubek-random", version = "0.2.0", default-features = false } -cubek-quant = { path = "../cubek-quant", version = "0.2.0" } +cubek-random = { package = "t4a-cubek-random", path = "../cubek-random", version = "0.2.0", default-features = false } +cubek-quant = { package = "t4a-cubek-quant", path = "../cubek-quant", version = "0.2.0" } bytemuck = { workspace = true } cubecl = { workspace = true, features = ["stdlib", "test-runtime"] } diff --git a/crates/cubek-test-utils/README.md b/crates/cubek-test-utils/README.md index b2c709cc1..e96bd1fdf 100644 --- a/crates/cubek-test-utils/README.md +++ b/crates/cubek-test-utils/README.md @@ -1,3 +1,5 @@ +> **tensor4all temporary fork.** This `t4a-cubek*` package is a temporary tensor4all fork used by `tenferro-rs` while required patches are being upstreamed to the official CubeK/CubeCL projects. It is not a replacement for upstream CubeK. + # cubek-test-utils Shared building blocks for kernel tests in CubeK: test-tensor builders, diff --git a/crates/cubek/Cargo.toml b/crates/cubek/Cargo.toml index 6be13f439..ae3d4643c 100644 --- a/crates/cubek/Cargo.toml +++ b/crates/cubek/Cargo.toml @@ -9,8 +9,9 @@ edition.workspace = true keywords = [] license.workspace = true name = "cubek" +publish = false readme.workspace = true -repository = "https://github.com/tracel-ai/cubek" +repository = "https://github.com/tensor4all/cubek" version.workspace = true [features] @@ -56,9 +57,9 @@ cubecl = { workspace = true } cubek-attention = { path = "../cubek-attention", version = "0.2.0", default-features = false, optional = true } cubek-interpolate = { path = "../cubek-interpolate", version = "0.2.0", default-features = false, optional = true } cubek-convolution = { path = "../cubek-convolution", version = "0.2.0", default-features = false, optional = true } -cubek-matmul = { path = "../cubek-matmul", version = "0.2.0", default-features = false, optional = true } -cubek-quant = { path = "../cubek-quant", version = "0.2.0", default-features = false, optional = true } -cubek-random = { path = "../cubek-random", version = "0.2.0", default-features = false, optional = true } +cubek-matmul = { package = "t4a-cubek-matmul", path = "../cubek-matmul", version = "0.2.0", default-features = false, optional = true } +cubek-quant = { package = "t4a-cubek-quant", path = "../cubek-quant", version = "0.2.0", default-features = false, optional = true } +cubek-random = { package = "t4a-cubek-random", path = "../cubek-random", version = "0.2.0", default-features = false, optional = true } cubek-reduce = { path = "../cubek-reduce", version = "0.2.0", default-features = false, optional = true } cubek-fft = { path = "../cubek-fft", version = "0.2.0", default-features = false, optional = true } -cubek-std = { path = "../cubek-std", version = "0.2.0", default-features = false, optional = true } +cubek-std = { package = "t4a-cubek-std", path = "../cubek-std", version = "0.2.0", default-features = false, optional = true } diff --git a/xtask/Cargo.toml b/xtask/Cargo.toml index 8e40f288e..b0d674ca1 100644 --- a/xtask/Cargo.toml +++ b/xtask/Cargo.toml @@ -2,6 +2,7 @@ name = "xtask" version = "0.2.0" edition.workspace = true +publish = false # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html