Skip to content

Commit 01716cf

Browse files
committed
build(speechlm2): add cut-cross-entropy to compiled extras
Original-Commit: 2c645f4 Signed-off-by: Piotr Żelasko <pzelasko@nvidia.com>
1 parent 2e41db0 commit 01716cf

2 files changed

Lines changed: 20 additions & 0 deletions

File tree

pyproject.toml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -245,6 +245,7 @@ cu13 = [
245245
compiled = [
246246
"onnx-ir==0.2.1",
247247
"onnxscript==0.7.0",
248+
"cut-cross-entropy==25.3.2",
248249
"deep_ep==1.2.1",
249250
"nv-grouped-gemm==1.1.4.post8",
250251
"causal-conv1d==1.6.2.post1",
@@ -258,6 +259,7 @@ compiled = [
258259
compiled-a100 = [
259260
"onnx-ir==0.2.1",
260261
"onnxscript==0.7.0",
262+
"cut-cross-entropy==25.3.2",
261263
"nv-grouped-gemm==1.1.4.post8",
262264
"causal-conv1d==1.6.2.post1",
263265
"mamba-ssm==2.3.2.post1",
@@ -475,6 +477,7 @@ deep-ep = ["setuptools"]
475477
# metadata is mirrored below so this repository controls the CUDA wheel index.
476478
[tool.uv.sources]
477479
nemo_automodel = { git = "https://github.com/NVIDIA-NeMo/Automodel.git", rev = "7b15c9bfbc620bec4dd3b819fce2e48726a7ddeb" }
480+
cut-cross-entropy = { git = "https://github.com/apple/ml-cross-entropy.git", rev = "87a86aba72cfd2f0d8abecaf81c13c4528ea07d8" }
478481
megatron-fsdp = { git = "https://github.com/yuhezhang-ai/Megatron-LM.git", rev = "455389c480af6b3acdca74c7830c68b3274eb083", subdirectory = "megatron/core/distributed/fsdp/src" }
479482
deep_ep = { git = "https://github.com/deepseek-ai/DeepEP.git", rev = "7febc6e25660af0f54d95dd781ecdcd62265ecca" }
480483
torch = [

uv.lock

Lines changed: 17 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)