From 91c2e8c2a99793b7c99cae1e15f6b681a2854994 Mon Sep 17 00:00:00 2001 From: dcvz Date: Sat, 6 Apr 2024 00:44:22 +0200 Subject: [PATCH 1/3] enable wgpu on linux --- .github/workflows/validate.yml | 3 --- 1 file changed, 3 deletions(-) diff --git a/.github/workflows/validate.yml b/.github/workflows/validate.yml index d177d18..deec729 100644 --- a/.github/workflows/validate.yml +++ b/.github/workflows/validate.yml @@ -35,9 +35,6 @@ jobs: # windows can have CPU Vulkan but Burn doesn't select CPU well yet - os: windows-latest backend: "wgpu" - # ubuntu is throwing SIGSEGV - - os: ubuntu-latest - backend: "wgpu" steps: - name: Checkout uses: actions/checkout@v2 From 6ecad2e2a1e42e0a164da0d3f85653dc77f41a71 Mon Sep 17 00:00:00 2001 From: dcvz Date: Sat, 6 Apr 2024 00:53:42 +0200 Subject: [PATCH 2/3] Disable test on wgpu linux CI --- src/transformers/clip.rs | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/transformers/clip.rs b/src/transformers/clip.rs index ca1ea0c..dcc524b 100644 --- a/src/transformers/clip.rs +++ b/src/transformers/clip.rs @@ -361,6 +361,12 @@ mod tests { #[test] fn test_init_text_embeddings() { + // This test is known to fail on WGPU using mesa on a Linux Github CI. + #[cfg(all(feature = "std", target_os = "linux", feature = "wgpu"))] + if std::env::var("CI").is_ok() { + return; + } + let device = Default::default(); let clip_config = ClipConfig::v1_5(); let text_embeddings: ClipTextEmbeddings = From 41b67d2081f09822d6ee9d6267b0c4bab9c7484d Mon Sep 17 00:00:00 2001 From: dcvz Date: Sat, 13 Apr 2024 22:12:34 +0200 Subject: [PATCH 3/3] Use blaze linux runner --- .github/workflows/validate.yml | 2 +- src/transformers/clip.rs | 6 ------ 2 files changed, 1 insertion(+), 7 deletions(-) diff --git a/.github/workflows/validate.yml b/.github/workflows/validate.yml index deec729..55f43ad 100644 --- a/.github/workflows/validate.yml +++ b/.github/workflows/validate.yml @@ -25,7 +25,7 @@ jobs: runs-on: ${{ matrix.os }} strategy: matrix: - os: [blaze/macos-14, ubuntu-latest, windows-latest] + os: [blaze/macos-14, blaze/ubuntu-latest, windows-latest] rust: [stable] backend: ["ndarray", "wgpu", "torch"] include: diff --git a/src/transformers/clip.rs b/src/transformers/clip.rs index dcc524b..ca1ea0c 100644 --- a/src/transformers/clip.rs +++ b/src/transformers/clip.rs @@ -361,12 +361,6 @@ mod tests { #[test] fn test_init_text_embeddings() { - // This test is known to fail on WGPU using mesa on a Linux Github CI. - #[cfg(all(feature = "std", target_os = "linux", feature = "wgpu"))] - if std::env::var("CI").is_ok() { - return; - } - let device = Default::default(); let clip_config = ClipConfig::v1_5(); let text_embeddings: ClipTextEmbeddings =