Commit 23b5013
authored
add multi_tensor_compute_scale_inv_e8m0 and change call erro (#74)
# Description
Added the binding and invocation for the
`multi_tensor_compute_scale_inv_e8m0` operator for the Kunlunxin vendor,
and corrected a syntax error in the invocation of the
`multi_tensor_compute_scale_and_scale_inv` operator.
Fixes # (issue)
## Type of change
- [ ] Documentation change (change only to the documentation, either a
fix or a new content)
- [1] Bug fix (non-breaking change which fixes an issue)
- [1] New feature (non-breaking change which adds functionality)
- [ ] Breaking change (fix or feature that would cause existing
functionality to not work as expected)
- [ ] Infra/Build change
- [ ] Code refactoring
## Changes
Added code for the binding and invocation of the
`multi_tensor_compute_scale_inv_e8m0` operator.
Modified the code calling `multi_tensor_compute_scale_and_scale_inv`.
# Checklist:
- [ 1] I have read and followed the [contributing
guidelines](https://github.com/NVIDIA/TransformerEngine/blob/main/CONTRIBUTING.rst)
- [ 1] The functionality is complete
- [ 1] I have commented my code, particularly in hard-to-understand
areas
- [ 1] I have made corresponding changes to the documentation
- [ 1] My changes generate no new warnings
- [ 1] I have added tests that prove my fix is effective or that my
feature works
- [ 1] New and existing unit tests pass locally with my changes1 parent 25e80e9 commit 23b5013
2 files changed
Lines changed: 24 additions & 1 deletion
Lines changed: 16 additions & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
357 | 357 | | |
358 | 358 | | |
359 | 359 | | |
360 | | - | |
| 360 | + | |
361 | 361 | | |
362 | 362 | | |
| 363 | + | |
| 364 | + | |
| 365 | + | |
| 366 | + | |
| 367 | + | |
| 368 | + | |
| 369 | + | |
| 370 | + | |
| 371 | + | |
| 372 | + | |
| 373 | + | |
| 374 | + | |
| 375 | + | |
| 376 | + | |
| 377 | + | |
Lines changed: 8 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
173 | 173 | | |
174 | 174 | | |
175 | 175 | | |
| 176 | + | |
| 177 | + | |
| 178 | + | |
| 179 | + | |
| 180 | + | |
| 181 | + | |
| 182 | + | |
| 183 | + | |
176 | 184 | | |
177 | 185 | | |
178 | 186 | | |
0 commit comments