OmpModule.lua


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

local OmpModule,parent = torch.class('nn.OmpModule','nn.Module')

function OmpModule:__init()
   parent.__init(self)
   if openmp then
      self.nThread = openmp.getNumThreads()
   else
      self.nThread = 1
   end
end

function OmpModule:read(file)
   parent.read(self, file)
   if openmp then
      self.nThread = openmp.getNumThreads()
   else
      self.nThread = 1
   end
end