From 73da28805bf6e6d616f3fb3d0a4d5720904e1f59 Mon Sep 17 00:00:00 2001
From: Maddox Werts <accounts@objnull.net>
Date: Mon, 17 Mar 2025 12:14:18 -0400
Subject: [PATCH] Updated WebGPU to WGPU

---
 project/src/neural.rs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/project/src/neural.rs b/project/src/neural.rs
index 20a8513..a834664 100644
--- a/project/src/neural.rs
+++ b/project/src/neural.rs
@@ -6,7 +6,7 @@ mod training;
 use super::config::OperationMode;
 
 use burn::{
-    backend::{Autodiff, WebGpu},
+    backend::{Autodiff, Wgpu},
     data::dataset::Dataset,
     optim::AdamConfig,
 };
@@ -26,7 +26,7 @@ impl NeuralNetwork {
 
     // Functions
     fn train(&self) {
-        type MyBackend = WebGpu<f32, i32>;
+        type MyBackend = Wgpu<f32, i32>;
         type MyAutodiffBackend = Autodiff<MyBackend>;
 
         // Create a default Wgpu device
@@ -52,7 +52,7 @@ impl NeuralNetwork {
         );
     }
     fn infer(&self) {
-        type MyBackend = WebGpu<f32, i32>;
+        type MyBackend = Wgpu<f32, i32>;
 
         let device = burn::backend::wgpu::WgpuDevice::default();