Meta-crates (Phase 2): - rtx-core / rtx-training / rtx-inference-stack gain cuda and metal features threading into their sub-crates; GPU was previously unreachable through the user-facing bundles. - rtx-training restores rtx-distributed (the hpc-channels blocker is gone) so the advertised DistributedTransformerTrainer resolves; drops the unused rtx-runtime dep. - rtx-transformers drops unused rtx-backend/rtx-backend-cpu deps (stale comment referenced a teacher that never used them). Never-compiled CUDA paths fixed (surfaced by the new feature wiring, verified on RTX 5060 Ti / CUDA 13.1): - rtx-compress build.rs: missing Path/Command/fs imports. - rtx-flash-attention flash_decode_forward: reborrow &mut kernel args. - rtx-transformers: rope kernel include path, cudarc 0.18 Arc<CudaModule>, PushKernelArg imports in jepa_gpu, edition-2024 ref patterns. - rtx-memory: full cudarc 0.18 port (CudaContext, stream-based alloc, DevicePtr accessors, error enum formatting) across gpu_pinning, gpu_transfer, gpu_real, gpu_allocator/arena, gpu_tests. JEPA (Phase 3): - JepaRunConfig::apply_cluster_plan consumes ClusterTrainingPlan (batch size, TP/DP, world size, total steps) so jepa_cluster is no longer standalone dead config; ViTSizeStr::approx_params_m feeds JepaParallelConfig::for_model_and_cluster. - WebDatasetShard::load reads real .tar shards from disk via the existing parser (gzip rejected explicitly); to_in_memory documented as synthetic/test-only. - New image-decode feature actually defines the dep for the previously unreachable cfg(feature = "image-decode") JPEG/PNG decode path. Co-Authored-By: Claude Fable 5 <[email protected]>
29 lines
933 B
TOML
29 lines
933 B
TOML
[package]
|
|
name = "rtx-core"
|
|
version = "1.0.0"
|
|
edition.workspace = true
|
|
rust-version = "1.92"
|
|
authors = ["RustyTorch Team"]
|
|
license = "MIT OR Apache-2.0"
|
|
repository = "https://github.com/rustytorch/rustytorch"
|
|
description = "RTX core tensor operations and runtime - Essential functionality only"
|
|
keywords = ["machine-learning", "tensor", "gpu", "cuda", "core"]
|
|
categories = ["science", "mathematics"]
|
|
|
|
[dependencies]
|
|
rtx-tensor = { path = "../../core/rtx-tensor", version = "1.0.0" }
|
|
rtx-runtime = { path = "../../core/rtx-runtime", version = "1.0.0" }
|
|
rtx-autograd = { path = "../../core/rtx-autograd", version = "1.0.0" }
|
|
rtx-memory = { path = "../../core/rtx-memory", version = "1.0.0" }
|
|
|
|
[features]
|
|
default = []
|
|
cuda = ["rtx-tensor/cuda", "rtx-runtime/cuda", "rtx-memory/cuda"]
|
|
metal = ["rtx-tensor/metal", "rtx-runtime/metal", "rtx-memory/metal"]
|
|
|
|
[dev-dependencies]
|
|
tokio-test = "0.4"
|
|
criterion = "0.5"
|
|
[lints]
|
|
workspace = true
|