Commit 19cfe11
[Bug Fix] Fix test that requre GPU (#1096)
SUMMARY:
Nightly tests uses gpus, ci tests dont.
In the set up of
`tests/llmcompressor/transformers/obcq/test_consecutive_runs.py`, a bug
was found where `quantization_config` was always passed. This will
result in an error if the model from the config file is not optimized.
We are passing in dense model, so it fails.
TEST PLAN:
Run the failing test, make sure it passes!
Signed-off-by: Rahul Tuli <rahul@neuralmagic.com>1 parent 6f02c91 commit 19cfe11
File tree
1 file changed
+6
-1
lines changed- tests/llmcompressor/transformers/obcq
1 file changed
+6
-1
lines changedLines changed: 6 additions & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
8 | 8 | | |
9 | 9 | | |
10 | 10 | | |
| 11 | + | |
11 | 12 | | |
12 | 13 | | |
13 | 14 | | |
| |||
137 | 138 | | |
138 | 139 | | |
139 | 140 | | |
| 141 | + | |
| 142 | + | |
| 143 | + | |
| 144 | + | |
| 145 | + | |
140 | 146 | | |
141 | 147 | | |
142 | 148 | | |
143 | | - | |
144 | 149 | | |
145 | 150 | | |
146 | 151 | | |
| |||
0 commit comments