Files
rustytorch/crates/meta/rtx-core/Cargo.toml
T
osobhandClaude Fable 5 1e3c604896 feat(meta,jepa): expose GPU features through meta-crates; wire JEPA cluster plan and real shard loading
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]>
2026-07-09 19:25:51 -07:00

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