prevent nn.GPU from switching devices of MultiCudaTensor

nicholas-leonard · Jul 4, 2016 · be944cf · be944cf
1 parent 26dbbfe
commit be944cf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/MultiCudaTensor.lua b/MultiCudaTensor.lua
@@ -6,6 +6,7 @@
 -- NCEModule requires.
 ------------------------------------------------------------------------
 local MCT = torch.class("torch.MultiCudaTensor")
+MCT.__noGPU__ = true -- will prevent nn.GPU from switching devices
 
 -- each buffer is indexed by device
 local buffers1, buffers2 = {}, {}