Commit 41c78df
authored
[NPU]Fix OOM in benchmark_tvd on NPU (#1103)
## Summary
`benchmark_tvd.py` can fail with an OOM error on Atlas 800I A2 (32GB)
because of exceeding the available NPU global memory.This change
addresses the issue and prevents the benchmark from exceeding device
memory capacity.
## Testing Done
Atlas 800I A2(32GB)
- [ ] run `make test` to ensure correctness
- [x] run `make checkstyle` to ensure code style
- [ ] run `make test-convergence` to ensure convergence1 parent 7644a0f commit 41c78df
1 file changed
+3
-1
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
114 | 114 | | |
115 | 115 | | |
116 | 116 | | |
117 | | - | |
| 117 | + | |
118 | 118 | | |
| 119 | + | |
| 120 | + | |
119 | 121 | | |
120 | 122 | | |
121 | 123 | | |
| |||
0 commit comments