Skip to content

Conversation

jeffkbkim
Copy link
Contributor

Summary:
TensorWeightedAvgMetric currently does not support FUSED_TASKS computation. With this patch, TWA supports FUSED_TASKS mode

Updated unit tests and created new ones for FUSED mode

Differential Revision: D77958663

@meta-cla meta-cla bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Aug 25, 2025
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D77958663

jeffkbkim added a commit to jeffkbkim/torchrec that referenced this pull request Aug 25, 2025
Summary:

TensorWeightedAvgMetric currently does not support FUSED_TASKS computation. With this patch, TWA supports FUSED_TASKS mode

Updated unit tests and created new ones for FUSED mode

Differential Revision: D77958663
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D77958663

jeffkbkim added a commit to jeffkbkim/torchrec that referenced this pull request Aug 25, 2025
Summary:

TensorWeightedAvgMetric currently does not support FUSED_TASKS computation. With this patch, TWA supports FUSED_TASKS mode

Updated unit tests and created new ones for FUSED mode

Differential Revision: D77958663
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D77958663

jeffkbkim added a commit to jeffkbkim/torchrec that referenced this pull request Aug 25, 2025
Summary:

TensorWeightedAvgMetric currently does not support FUSED_TASKS computation. With this patch, TWA supports FUSED_TASKS mode

Updated unit tests and created new ones for FUSED mode

Differential Revision: D77958663
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D77958663

jeffkbkim added a commit to jeffkbkim/torchrec that referenced this pull request Aug 25, 2025
Summary:

TensorWeightedAvgMetric currently does not support FUSED_TASKS computation. With this patch, TWA supports FUSED_TASKS mode

Updated unit tests and created new ones for FUSED mode

Differential Revision: D77958663
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D77958663

jeffkbkim added a commit to jeffkbkim/torchrec that referenced this pull request Aug 26, 2025
Summary:

TensorWeightedAvgMetric currently does not support FUSED_TASKS computation. With this patch, TWA supports FUSED_TASKS mode

Updated unit tests and created new ones for FUSED mode

Differential Revision: D77958663
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D77958663

jeffkbkim added a commit to jeffkbkim/torchrec that referenced this pull request Aug 26, 2025
Summary:
Pull Request resolved: meta-pytorch#3317

TensorWeightedAvgMetric currently does not support FUSED_TASKS computation. With this patch, TWA supports FUSED_TASKS mode

Updated unit tests and created new ones for FUSED mode

Differential Revision: D77958663
jeffkbkim added a commit to jeffkbkim/torchrec that referenced this pull request Aug 26, 2025
Summary:

TensorWeightedAvgMetric currently does not support FUSED_TASKS computation. With this patch, TWA supports FUSED_TASKS mode

Updated unit tests and created new ones for FUSED mode

Differential Revision: D77958663
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D77958663

@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D77958663

jeffkbkim added a commit to jeffkbkim/torchrec that referenced this pull request Aug 27, 2025
Summary:

TensorWeightedAvgMetric currently does not support FUSED_TASKS computation. With this patch, TWA supports FUSED_TASKS mode

Updated unit tests and created new ones for FUSED mode

Differential Revision: D77958663
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D77958663

jeffkbkim added a commit to jeffkbkim/torchrec that referenced this pull request Aug 28, 2025
Summary:

TensorWeightedAvgMetric currently does not support FUSED_TASKS computation. With this patch, TWA supports FUSED_TASKS mode

Updated unit tests and created new ones for FUSED mode

Reviewed By: iamzainhuda

Differential Revision: D77958663
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D77958663

jeffkbkim added a commit to jeffkbkim/torchrec that referenced this pull request Aug 28, 2025
Summary:

TensorWeightedAvgMetric currently does not support FUSED_TASKS computation. With this patch, TWA supports FUSED_TASKS mode

Updated unit tests and created new ones for FUSED mode

Reviewed By: iamzainhuda

Differential Revision: D77958663
Summary:
Pull Request resolved: meta-pytorch#3317

TensorWeightedAvgMetric currently does not support FUSED_TASKS computation. With this patch, TWA supports FUSED_TASKS mode

Updated unit tests and created new ones for FUSED mode

Reviewed By: iamzainhuda

Differential Revision: D77958663
jeffkbkim added a commit to jeffkbkim/torchrec that referenced this pull request Aug 28, 2025
Summary:

TensorWeightedAvgMetric currently does not support FUSED_TASKS computation. With this patch, TWA supports FUSED_TASKS mode

Updated unit tests and created new ones for FUSED mode

Reviewed By: iamzainhuda

Differential Revision: D77958663
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D77958663

1 similar comment
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D77958663

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants