From de3a73632a332a6e214a500ceba34d1f9d273d85 Mon Sep 17 00:00:00 2001
From: Guillaume Lagrange <lagrange.guillaume.1@gmail.com>
Date: Thu, 25 Apr 2024 12:04:30 -0400
Subject: [PATCH] Remove fusion backend

---
 bert-burn/Cargo.toml                  | 2 +-
 bert-burn/examples/infer-embedding.rs | 3 +--
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/bert-burn/Cargo.toml b/bert-burn/Cargo.toml
index fab1bd4..3da749f 100644
--- a/bert-burn/Cargo.toml
+++ b/bert-burn/Cargo.toml
@@ -2,7 +2,7 @@
 authors = ["Aasheesh Singh aasheeshdtu@gmail.com"]
 license = "MIT OR Apache-2.0"
 name = "bert-burn"
-version = "0.1.0"
+version = "0.2.0"
 edition = "2021"
 
 [features]
diff --git a/bert-burn/examples/infer-embedding.rs b/bert-burn/examples/infer-embedding.rs
index e4e714f..524f8ae 100644
--- a/bert-burn/examples/infer-embedding.rs
+++ b/bert-burn/examples/infer-embedding.rs
@@ -133,10 +133,9 @@ mod tch_cpu {
 mod wgpu {
     use crate::{launch, ElemType};
     use burn::backend::wgpu::{AutoGraphicsApi, Wgpu, WgpuDevice};
-    use burn::backend::Fusion;
 
     pub fn run() {
-        launch::<Fusion<Wgpu<AutoGraphicsApi, ElemType, i32>>>(WgpuDevice::default());
+        launch::<Wgpu<AutoGraphicsApi, ElemType, i32>>(WgpuDevice::default());
     }
 }