mlcube
mlcube copied to clipboard
TorchBench: Failing unit tests
There are 4 unit tests which are failing.
Failing Unit Tests:
test_train[pytorch_struct-cpu-jit] test_eval[pytorch_struct-cpu-jit] test_train[pytorch_struct-cpu-eager] test_eval[pytorch_struct-cpu-eager]
Follow up: What is the error?
Per @davidjurado this is fixed, and there is a PR in the pytorch repo. Next steps to share this with Xu (FB) - PR reviewer.
https://github.com/pytorch/benchmark/pull/559
We still need to:
- Make sure GPU tests pass
- Compare results when running in MLCube vs. native.