pytorch
a111dd90 - [dynamo] support comparing numpy ndarray (#91870)

Commit
2 years ago
[dynamo] support comparing numpy ndarray (#91870) The output of Torchbench model `doctr_det_predictor` on CPU is a `numpy ndarray`. When running the accuracy benchmark of this model, the below error is raised: `RuntimeError: unsupported type: ndarray`. Repro CMD: ```bash python benchmarks/dynamo/torchbench.py --accuracy --float32 -dcpu -n50 --inductor --no-skip --dashboard --only doctr_det_predictor --batch_size 1 --threads 1 ``` This PR adds the support to compare `numpy ndarray` in the dynamo utils. Pull Request resolved: https://github.com/pytorch/pytorch/pull/91870 Approved by: https://github.com/jgong5, https://github.com/Chillee
Author
Committer
Parents
Loading