pytorch - ✅(Solved) Fix DISABLED test_errors_nn_Embedding_xpu_float32 (__main__.TestModuleXPU) [2 pull requests, 1 comments, 2 participants]

Official PRs (…)
ON THIS PAGE

Recommended Tools

×6

Utilities matched from this issue’s tags and category — try them while you read without losing context.

GitHub issue graph ai analysis

Paste a GitHub issue URL. We fetch that issue, discover linked issues from bodies/comments/timeline, collect linked pull requests, and produce a structured English report.

The report is written in English Markdown for sharing and archival.

Helpful · Quick feedback

Loading…
GitHub stats
pytorch/pytorch#178762Fetched 2026-04-08 01:52:21
View on GitHub
Comments
1
Participants
2
Timeline
23
Reactions
0
Author
Timeline (top)
labeled ×6mentioned ×5subscribed ×5referenced ×3

Root Cause

This test was disabled because it is failing on main branch (recent examples).

PR fix notes

PR #178987: [xpu][fix] Fix nn.Embedding module failures on XPU

Description (problem / solution / changelog)

Stack from ghstack (oldest at bottom):

  • -> #178987

Motivation

https://github.com/pytorch/pytorch/pull/174180 introduces nn.Embedding module, which causes CI failures on XPU.

Additional Context

fix https://github.com/pytorch/pytorch/issues/178854 fix https://github.com/pytorch/pytorch/issues/178762

Changed files

  • torch/testing/_internal/common_modules.py (modified, +1/-2)
RAW_BUFFERClick to expand / collapse

Platforms: xpu

This test was disabled because it is failing on main branch (recent examples).

cc @mruberry @gujinghui @EikanWang @fengyuan14

extent analysis

Fix Plan

The fix involves re-enabling and updating the test case test_errors_nn_Embedding_xpu_float32 in test_modules.py to ensure compatibility with the XPU platform.

Steps to Fix

  • Re-enable the test case by removing any disabling decorators or comments.
  • Update the test case to handle XPU-specific requirements, such as data type conversions or platform-specific configurations.
  • Verify that the Embedding module works correctly with float32 data type on XPU.

Example Code

import torch
import torch.nn as nn
import torch_xpu

class TestModuleXPU:
    def test_errors_nn_Embedding_xpu_float32(self):
        # Create an Embedding module
        embedding = nn.Embedding(10,

Vote matrix · Quick signals

Works
Did the solution work? Tap to confirm.
Easy Fix
Was it a quick fix?
Time Saver
Did it save you time?
Blocking
Was it severely blocking?
Common Issue
Are others likely hitting this too?
Flaky / Intermittent
Is it intermittent?
Verified / Reproducible
Can you reproduce it reliably?
Loading…

Still need to ship something?

×6

Another batch ranked right after the header list — different links, same matching logic.

Back to top recommendations

TRENDING

pytorch - ✅(Solved) Fix DISABLED test_errors_nn_Embedding_xpu_float32 (__main__.TestModuleXPU) [2 pull requests, 1 comments, 2 participants]