Code

Update ort CIs (slow, gpu, train)#2024

Code

Merged

echarlaix merged 73 commits intomainhuggingface/optimum:mainfrom enable-ort-gpu-testshuggingface/optimum:enable-ort-gpu-testsCopy head branch name to clipboardJan 29, 2025

Commits

Commits on Sep 14, 2024

• update ort CIs

  IlyasMoutawwakil

  committed
  17bc171

  Copy full SHA for 17bc171

• fix train ci

  IlyasMoutawwakil

  committed
  fbaa980

  Copy full SHA for fbaa980

• fix gpu ci

  IlyasMoutawwakil

  committed
  90aa85d

  Copy full SHA for 90aa85d

• gpus all

  IlyasMoutawwakil

  committed
  87c9f3e

  Copy full SHA for 87c9f3e

• devel

  IlyasMoutawwakil

  committed
  0c1c6bd

  Copy full SHA for 0c1c6bd

Commits on Sep 15, 2024

• enable trt

  IlyasMoutawwakil

  committed
  430260e

  Copy full SHA for 430260e

• fix

  IlyasMoutawwakil

  committed
  00e51c7

  Copy full SHA for 00e51c7

• fix

  IlyasMoutawwakil

  committed
  3fc5486

  Copy full SHA for 3fc5486

Commits on Sep 16, 2024

• fix

  IlyasMoutawwakil

  committed
  8044232

  Copy full SHA for 8044232

Commits on Sep 27, 2024

• test

  IlyasMoutawwakil

  committed
  2fd4d47

  Copy full SHA for 2fd4d47

• rename

  IlyasMoutawwakil

  committed
  1f322fc

  Copy full SHA for 1f322fc

• change instance

  IlyasMoutawwakil

  committed
  6f7c599

  Copy full SHA for 6f7c599

• test

  IlyasMoutawwakil

  committed
  806faca

  Copy full SHA for 806faca

Commits on Sep 28, 2024

• use available

  IlyasMoutawwakil

  committed
  3eecee6

  Copy full SHA for 3eecee6

Commits on Dec 10, 2024

• Merge branch 'main' into enable-ort-gpu-tests

  IlyasMoutawwakil

  authored
  ab62319

  Copy full SHA for ab62319

Commits on Jan 10, 2025

• Merge branch 'main' into enable-ort-gpu-tests

  IlyasMoutawwakil

  committed
  1b7e652

  Copy full SHA for 1b7e652

• update

  IlyasMoutawwakil

  committed
  cebe6bf

  Copy full SHA for cebe6bf

• shorter labels as well

  IlyasMoutawwakil

  committed
  d0f62b0

  Copy full SHA for d0f62b0

• add onnxruntime-traning

  IlyasMoutawwakil

  committed
  d001b9b

  Copy full SHA for d001b9b

Commits on Jan 13, 2025

• Merge branch 'main' into enable-ort-gpu-tests

  IlyasMoutawwakil

  authored
  d271637

  Copy full SHA for d271637

• fix onnxruntime package checking

  IlyasMoutawwakil

  committed
  a318c0a

  Copy full SHA for a318c0a

• Merge branch 'enable-ort-gpu-tests' of https://github.com/huggingface/optimum into enable-ort-gpu-tests

  IlyasMoutawwakil

  committed
  7597692

  Copy full SHA for 7597692

• fix typo

  IlyasMoutawwakil

  committed
  a6b3a8e

  Copy full SHA for a6b3a8e

• fix typo

  IlyasMoutawwakil

  committed
  a5c76c4

  Copy full SHA for a5c76c4

• remove torch version

  IlyasMoutawwakil

  committed
  745ad8d

  Copy full SHA for 745ad8d

• fix trainer

  IlyasMoutawwakil

  committed
  bb48c4d

  Copy full SHA for bb48c4d

• fixed trt ep by using trt docker image (the only way to make sure everything works)

  IlyasMoutawwakil

  committed
  0518dfd

  Copy full SHA for 0518dfd

• latest trt version

  IlyasMoutawwakil

  committed
  9635ec4

  Copy full SHA for 9635ec4

• remove pkv speedup timing since never used

  IlyasMoutawwakil

  committed
  cb9cb7f

  Copy full SHA for cb9cb7f

• trust remote code for training datasets

  IlyasMoutawwakil

  committed
  eb25460

  Copy full SHA for eb25460

• remove rocm from diffusers tests

  IlyasMoutawwakil

  committed
  0a7a23d

  Copy full SHA for 0a7a23d

• move ort training tests to onnxruntime-training

  IlyasMoutawwakil

  committed
  64e9c86

  Copy full SHA for 64e9c86

Commits on Jan 14, 2025

• fix ort training

  IlyasMoutawwakil

  committed
  bbed6bc

  Copy full SHA for bbed6bc

• fix

  IlyasMoutawwakil

  committed
  1334200

  Copy full SHA for 1334200

• style

  IlyasMoutawwakil

  committed
  84bf7ee

  Copy full SHA for 84bf7ee

• always assert closenes and not equality

  IlyasMoutawwakil

  committed
  be10d26

  Copy full SHA for be10d26

• fixed perceiver

  IlyasMoutawwakil

  committed
  7ba72a6

  Copy full SHA for 7ba72a6

• fixed missing position ids when attn mask is given

  IlyasMoutawwakil

  committed
  eceba5b

  Copy full SHA for eceba5b

• remove num_labels from output shapes as it's not a dynamic axis

  IlyasMoutawwakil

  committed
  9150e05

  Copy full SHA for 9150e05

• raise error on missing mandatory inputs

  IlyasMoutawwakil

  committed
  198ce06

  Copy full SHA for 198ce06

• added atol and rtol as part of the ORTModelTestMixin class

  IlyasMoutawwakil

  committed
  930103f

  Copy full SHA for 930103f

• fix segformer image segmentation

  IlyasMoutawwakil

  committed
  49cfdc0

  Copy full SHA for 49cfdc0

• style

  IlyasMoutawwakil

  committed
  5b8efd4

  Copy full SHA for 5b8efd4

• fix vision encoder io binding

  IlyasMoutawwakil

  committed
  941484a

  Copy full SHA for 941484a

Commits on Jan 15, 2025

• hot fix io binding, remove its dependency to the order of inputs and make sure it's actually being tested

  IlyasMoutawwakil

  committed
  18e887d

  Copy full SHA for 18e887d

• fix

  IlyasMoutawwakil

  committed
  88a7e8b

  Copy full SHA for 88a7e8b

• typo

  IlyasMoutawwakil

  committed
  e9abe6a

  Copy full SHA for e9abe6a

• unify io binding api with non io binding

  IlyasMoutawwakil

  committed
  c9b45ee

  Copy full SHA for c9b45ee

• force evaluated shape to int

  IlyasMoutawwakil

  committed
  aad9aaf

  Copy full SHA for aad9aaf

• mark pix2struct io binding tests

  IlyasMoutawwakil

  committed
  a29706e

  Copy full SHA for a29706e

Commits on Jan 16, 2025

• force contiguity in forward pass

  IlyasMoutawwakil

  committed
  821c997

  Copy full SHA for 821c997

• fixed cryptic contiguity problems

  IlyasMoutawwakil

  committed
  cc2e124

  Copy full SHA for cc2e124

• fix some

  IlyasMoutawwakil

  committed
  3a2bcee

  Copy full SHA for 3a2bcee

• fix vision2seq modeling and testing

  IlyasMoutawwakil

  committed
  f0ea288

  Copy full SHA for f0ea288

Commits on Jan 28, 2025

• Merge branch 'main' into enable-ort-gpu-tests

  IlyasMoutawwakil

  authored
  7e122c0

  Copy full SHA for 7e122c0

• Update setup.py

  IlyasMoutawwakil

  authored
  dc2361d

  Copy full SHA for dc2361d

• update import utils

  IlyasMoutawwakil

  committed
  4eb95f1

  Copy full SHA for 4eb95f1

• Update optimum/onnxruntime/modeling_ort.py

  IlyasMoutawwakil

  authored
  7f1fc40

  Copy full SHA for 7f1fc40

• fix vision encoder decoder io binding

  IlyasMoutawwakil

  committed
  696cc95

  Copy full SHA for 696cc95

• enable bigbird and bigbirg pegasus and seperate timm slow tests to untangle them

  IlyasMoutawwakil

  committed
  1827450

  Copy full SHA for 1827450

• use bigger machine for slow tests

  IlyasMoutawwakil

  committed
  41abf7f

  Copy full SHA for 41abf7f

• lower atol and rtol for image classification logits

  IlyasMoutawwakil

  committed
  6f3084a

  Copy full SHA for 6f3084a

• fix

  IlyasMoutawwakil

  committed
  010030e

  Copy full SHA for 010030e

• large

  IlyasMoutawwakil

  committed
  445b291

  Copy full SHA for 445b291

Commits on Jan 29, 2025

• enable more Longformer and MCTCT

  IlyasMoutawwakil

  committed
  04c8904

  Copy full SHA for 04c8904

• enable commented models in export as well

  IlyasMoutawwakil

  committed
  18e1844

  Copy full SHA for 18e1844

• uncomment timm slow models, big bird optimization and marian pkv comparison

  IlyasMoutawwakil

  committed
  4487c74

  Copy full SHA for 4487c74

• Merge branch 'main' into enable-ort-gpu-tests

  IlyasMoutawwakil

  authored
  24d682e

  Copy full SHA for 24d682e

• Merge branch 'main' into enable-ort-gpu-tests

  IlyasMoutawwakil

  authored
  def5fdb

  Copy full SHA for def5fdb

• fix whisper/speech_to_text test and make convolution deterministic

  IlyasMoutawwakil

  committed
  458355d

  Copy full SHA for 458355d

• pin torch for ort training

  IlyasMoutawwakil

  committed
  881015c

  Copy full SHA for 881015c

• ctc and speech also uses convolution so has to be deterministic

  IlyasMoutawwakil

  committed
  7c8c56f

  Copy full SHA for 7c8c56f

• revert vison2seq atol

  IlyasMoutawwakil

  committed
  3a4bac9

  Copy full SHA for 3a4bac9