add isCUDA() on Type

Author: killeent

templates/Type.h

@@ -88,6 +88,7 @@ struct Type {
   virtual ~Type() {}
   virtual ScalarType scalarType() = 0;
   virtual Backend backend() = 0;
+  virtual bool isCuda() = 0;
   virtual bool isSparse() = 0;
   virtual bool isDistributed() = 0;
   static void registerAll(Context * context);

templates/TypeDerived.cpp

@@ -19,6 +19,7 @@ ScalarType ${Type}::scalarType() {
 Backend ${Type}::backend() {
   return Backend::${Backend};
 }
+bool ${Type}::isCuda() { return backend() == kCUDA; }
 bool ${Type}::isSparse() { return false; }
 bool ${Type}::isDistributed() { return false; }
 

templates/TypeDerived.h

@@ -10,6 +10,7 @@ struct ${Type} : public Type {
   explicit ${Type}(Context* context);
   virtual ScalarType scalarType() override;
   virtual Backend backend() override;
+  virtual bool isCuda() override;
   virtual bool isSparse() override;
   virtual bool isDistributed() override;
   virtual std::unique_ptr<Storage> storage() override;