2 Commits

Author	SHA1	Message	Date
Anton Lozhkov	4e59bcc680	[CI] Framework and hardware-specific CI tests (#997) ... * [WIP][CI] Framework and hardware-specific docker images for CI tests * username * fix cpu * try out the image * push latest * update workspace * no root isolation for actions * add a flax image * flax and onnx matrix * fix runners * add reports * onnxruntime image * retry tpu * fix * fix * build onnxruntime * naming * onnxruntime-gpu image * onnxruntime-gpu image, slow tests * latest jax version * trigger flax * run flax tests in one thread * fast flax tests on cpu * fast flax tests on cpu * trigger slow tests * rebuild torch cuda * force cuda provider * fix onnxruntime tests * trigger slow * don't specify gpu for tpu * optimize * memory limit * fix flax tests * disable docker cache	2022-11-02 14:07:07 +01:00
Kashif Rasul	240abddfbc	[Flax] added broadcast_to_shape_from_left helper and Scheduler tests (#864) ... * added broadcast_to_shape_from_left helper * initial tests * fixed pndm tests * shape required for pndm * added require_flax * fix style * fix more imports Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>	2022-10-25 13:43:24 +02:00