Commit 03dc386
authored
Update onnxruntime-gpu (#697)
## What does this PR do?
**Type of change:** Bug fix
**Overview:** Updated setup.py to use only onnxruntime-gpu and removed
onnxruntime-directml as dependency.
Also changed onnxruntime-gpu version in examples.
## Testing
Tested int4 quantization and MMLU benchmark with updated onnxruntime-gpu
, working as expected
---------
Signed-off-by: Hrishith Thadicherla <hthadicherla@nvidia.com>1 parent cb34335 commit 03dc386
3 files changed
Lines changed: 3 additions & 3 deletions
File tree
- examples/windows/onnx_ptq/whisper
- tests/unit/torch/quantization
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
5 | 5 | | |
6 | 6 | | |
7 | 7 | | |
8 | | - | |
| 8 | + | |
9 | 9 | | |
10 | 10 | | |
11 | 11 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
50 | 50 | | |
51 | 51 | | |
52 | 52 | | |
53 | | - | |
| 53 | + | |
54 | 54 | | |
55 | 55 | | |
56 | 56 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
38 | 38 | | |
39 | 39 | | |
40 | 40 | | |
41 | | - | |
| 41 | + | |
42 | 42 | | |
43 | 43 | | |
44 | 44 | | |
0 commit comments