Revert "update to_gpu"

This reverts commit a8489bb.
konas122 · May 4, 2024 · 6182a28 · 6182a28
1 parent 5d3c365
commit 6182a28
Show file tree

Hide file tree

Showing 11 changed files with 1 addition and 979 deletions.
diff --git a/dazero/core.py b/dazero/core.py
@@ -3,7 +3,6 @@
 import numpy as np
 
 import dazero
-import dazero.cuda
 
 
 class Config:
@@ -174,8 +173,6 @@ def to_cpu(self):
             self.data = dazero.cuda.as_numpy(self.data)
 
     def to_gpu(self):
-        if dazero.cuda.gpu_enable == False:
-            return
         if self.data is not None:
             self.data = dazero.cuda.as_cupy(self.data)
 

diff --git a/dazero/dataloaders.py b/dazero/dataloaders.py
@@ -60,9 +60,6 @@ def to_cpu(self):
         self.gpu = False
 
     def to_gpu(self):
-        if cuda.gpu_enable == False:
-            self.gpu = False
-            return
         self.gpu = True
 
 

diff --git a/dazero/layers.py b/dazero/layers.py
@@ -45,8 +45,6 @@ def to_cpu(self):
             param.to_cpu()
 
     def to_gpu(self):
-        if cuda.gpu_enable == False:
-            return
         for param in self.params():
             param.to_gpu()
 

diff --git a/examples/Diffusion/diffusion.ipynb b/examples/Diffusion/diffusion.ipynb
diff --git a/examples/Diffusion/modules/diffusion.py b/examples/Diffusion/modules/diffusion.py
diff --git a/examples/Diffusion/modules/sampler.py b/examples/Diffusion/modules/sampler.py
diff --git a/examples/Diffusion/modules/trainer.py b/examples/Diffusion/modules/trainer.py
diff --git a/examples/Diffusion/modules/unet.py b/examples/Diffusion/modules/unet.py