Commit 9aa7e20
authored
[fix] fix plugin te_general_grouped test bug (#81)
## Summary
Replace the grouped GEMM plugin test import from
`transformer_engine_torch_nv` to `transformer_engine_torch`.
## Why
The test should load the standard Transformer Engine torch extension
module. Importing `transformer_engine_torch_nv` can fail because that is
not the expected module name in this project.
## Impact
This fixes startup/import failure for
`transformer_engine/plugin/tests/test_te_general_grouped.py` before the
grouped GEMM checks run.
## Validation
- `git diff --check origin/main...HEAD`
- `python3 -c 'import ast, pathlib; path =
pathlib.Path("transformer_engine/plugin/tests/test_te_general_grouped.py");
ast.parse(path.read_text())'`
Full test execution was not run locally because it requires the project
runtime/CUDA environment.1 parent b7f65d1 commit 9aa7e20
1 file changed
Lines changed: 1 addition & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
22 | 22 | | |
23 | 23 | | |
24 | 24 | | |
25 | | - | |
| 25 | + | |
26 | 26 | | |
27 | 27 | | |
28 | 28 | | |
| |||
0 commit comments