Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Dgalvez/wip conditional nodes functional #9212

Conversation

tingyangk
Copy link

What does this PR do ?

Add my_torch_cond function, which emulates torch.cond. Also add support for CUDA Graphs stream capture to my_torch_cond.

This is to support easier implementation of RNN-T greedy inference, and other data-dependent control flow heavy algorithms.

Collection: Core, ASR

This is a draft for now. No need for the core nemo people to review it.

@github-actions github-actions bot added core Changes to NeMo Core ASR labels May 15, 2024
Copy link
Contributor

This PR is stale because it has been open for 14 days with no activity. Remove stale label or comment or update or this will be closed in 7 days.

@github-actions github-actions bot added the stale label Jun 12, 2024
@tingyangk tingyangk force-pushed the dgalvez/WIP-conditional-nodes-functional branch from 975a806 to 8aa5080 Compare June 14, 2024 01:02
@github-actions github-actions bot removed the stale label Jul 26, 2024
Copy link
Contributor

github-actions bot commented Aug 9, 2024

This PR is stale because it has been open for 14 days with no activity. Remove stale label or comment or update or this will be closed in 7 days.

@github-actions github-actions bot added the stale label Aug 9, 2024
Copy link
Contributor

This PR was closed because it has been inactive for 7 days since being marked as stale.

@github-actions github-actions bot closed this Aug 17, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
ASR core Changes to NeMo Core stale
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants