You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Using the example with --device cpu leads to crash:
python inference/inference_sim.py -a resnet50 -b 512 -pcq_w -pcq_a -sh --qtype int4 -qw int4 --device cpu
The bug comes from calling torch.cuda.clear().
Likewise when the program finishes.
cuda device should not be called in cpu mode.
The text was updated successfully, but these errors were encountered:
Using the example with --device cpu leads to crash:
python inference/inference_sim.py -a resnet50 -b 512 -pcq_w -pcq_a -sh --qtype int4 -qw int4 --device cpu
The bug comes from calling torch.cuda.clear().
Likewise when the program finishes.
cuda device should not be called in cpu mode.
The text was updated successfully, but these errors were encountered: