Fix a logic error in PFE (LLNL#2029)

benson31 · Jan 19, 2022 · 8714c40 · 8714c40
1 parent 72e5ac9
commit 8714c40
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/lbann/core/operator_layers.py b/python/lbann/core/operator_layers.py
@@ -53,7 +53,7 @@ def export_proto(self):
                 device = lbann.DeviceAllocation.CPU
             elif self.device.lower() == 'gpu':
                 device = lbann.DeviceAllocation.GPU
-        else:
+        elif self.device is not None:
             raise TypeError('Unknown type for field device ' + str(type(device)))
 
         # Configure operators to match layer