[Util] Introduce ContentHashClass.

CHANGELOG.md

@@ -31,6 +31,8 @@ List of major changes and improvements
 
   - Use a single global argument parser.
 
+  - Introduce ContentHashClass.
+
 
 ### Fixed
 

nn_dataflow/Layer.py

@@ -18,9 +18,9 @@
 program. If not, see <https://opensource.org/licenses/BSD-3-Clause>.
 """
 
-from .Util import StringifyClass
+from .Util import StringifyClass, ContentHashClass
 
-class Layer(StringifyClass):
+class Layer(StringifyClass, ContentHashClass):
     '''
     Base NN layer.
 
@@ -122,17 +122,6 @@ def total_ops(self, batch_size=1):
         ''' Get total number of operations. '''
         return self.total_ofmap_size() * self.ops_per_neuron() * batch_size
 
-    def __eq__(self, other):
-        if isinstance(other, self.__class__):
-            return self.__dict__ == other.__dict__
-        return NotImplemented
-
-    def __ne__(self, other):
-        return not self.__eq__(other)
-
-    def __hash__(self):
-        return hash(tuple(sorted(self.__dict__.items())))
-
 
 class InputLayer(Layer):
     '''

nn_dataflow/PartitionScheme.py

@@ -24,9 +24,9 @@
 from . import Util
 from .Layer import ConvLayer, LocalRegionLayer
 from .PhyDim2 import PhyDim2
-from .Util import StringifyClass
+from .Util import StringifyClass, ContentHashClass
 
-class PartitionScheme(StringifyClass):
+class PartitionScheme(StringifyClass, ContentHashClass):
     '''
     A parallel processing partitioning scheme.
     '''
@@ -123,14 +123,3 @@ def part_layer(self, layer, batch_size):
 
         return p_layer, p_batch_size, p_occ
 
-    def __eq__(self, other):
-        if isinstance(other, self.__class__):
-            return self.__dict__ == other.__dict__
-        return NotImplemented
-
-    def __ne__(self, other):
-        return not self.__eq__(other)
-
-    def __hash__(self):
-        return hash(tuple(sorted(self.__dict__.items())))
-

nn_dataflow/Util.py

@@ -37,6 +37,26 @@ def __str__(self):
                        in sorted(self.__dict__.items())]))
 
 
+class ContentHashClass(object):
+    '''
+    Class using the content instead of the object ID for hash.
+
+    Such class instance can be used as key in dictionary.
+    '''
+    # pylint: disable=too-few-public-methods
+
+    def __eq__(self, other):
+        if isinstance(other, self.__class__):
+            return self.__dict__ == other.__dict__
+        return NotImplemented
+
+    def __ne__(self, other):
+        return not self.__eq__(other)
+
+    def __hash__(self):
+        return hash(tuple(sorted(self.__dict__.items())))
+
+
 def idivc(valx, valy):
     '''
     Integer division and ceiling.