mirror of
https://github.com/nod-ai/AMD-SHARK-Studio.git
synced 2026-04-03 03:00:17 -04:00
dcc1bbac14
Added pytests-parametrize with device and dynamic options, also added worker flag in pytest to run tests concurrently.
15 lines
166 B
Plaintext
15 lines
166 B
Plaintext
-f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
|
|
--pre
|
|
|
|
numpy
|
|
torch
|
|
torchvision
|
|
|
|
# Build requirements.
|
|
pybind11
|
|
wheel
|
|
ninja
|
|
pytest
|
|
pytest-parallel
|
|
|