commit 7b9b538aa55c57c660dc74d506e5830a027082bd
author Trevor Killeen <killeentm@gmail.com> Fri Jul 14 10:39:40 2017 -0700
committer Edward Z. Yang <ezyang@mit.edu> Thu Nov 02 19:53:36 2017 -0400
tree fbc78f152c0620bc05656ef8f1497d6957db90ce
parent 95e4c4ff87793defd15091a954af868626d0105a

operator== for type

aten/src/ATen/templates/Type.cpp

diff --git a/aten/src/ATen/templates/Type.cpp b/aten/src/ATen/templates/Type.cpp
index 76480bb..609a755 100644
--- a/aten/src/ATen/templates/Type.cpp
+++ b/aten/src/ATen/templates/Type.cpp
@@ -54,6 +54,10 @@
   return tensor({}).fill_(s);
 }
 
+bool Type::operator==(const Type& other) const {
+  return this->ID() == other.ID();
+}
+
 ${type_method_definitions}
 
 }

aten/src/ATen/templates/Type.h

diff --git a/aten/src/ATen/templates/Type.h b/aten/src/ATen/templates/Type.h
index 994a5f1..0bdbbe9 100644
--- a/aten/src/ATen/templates/Type.h
+++ b/aten/src/ATen/templates/Type.h
@@ -111,6 +111,9 @@
   Tensor tensorFromBlob(void * data, IntList sizes);
   Tensor tensorFromBlob(void * data, IntList sizes, IntList strides);
   Tensor scalarTensor(Scalar s);
+
+  bool operator==(const Type& other) const;
+
   // example
   // virtual Tensor * add(Tensor & a, Tensor & b) = 0;
   ${type_method_declarations}