Add error messages for int64 non-available ops (#80)

* Add error messages for int64 non-available ops

* Move warning to common code

Author: abhudev

aten/src/ATen/native/mps/operations/Activation.mm

@@ -416,6 +416,7 @@ Tensor relu_mps(const Tensor& self) {
   using namespace mps;
   using CachedGraph = MPSUnaryCachedGraph;
   TORCH_CHECK(output.is_mps());
+  TORCH_CHECK(self.scalar_type() != ScalarType::Long, "MPS does not support sigmoid op with int64 input")
 
   if(output.numel() == 0) {
     return;

aten/src/ATen/native/mps/operations/BinaryOps.mm

@@ -27,6 +27,7 @@
 void binaryOpTensor(const Tensor& self, const Tensor& other, const Scalar& alpha,
                     const Tensor& output_, std::string op_name, BinaryOpBlock binaryBlock)
 {
+
   // it's possible to receive empty tensors here
   if (self.numel() == 0 || other.numel() == 0) {
     return;
@@ -203,6 +204,9 @@ void add_sub_template(const Tensor& self, const Tensor& other, const Scalar& alp
 
 #define CREATE_MPS_STRUCTURED_BINARY_OP_FUNC(func_out, func_stub, other_type)                   \
 TORCH_IMPL_FUNC(func_out) (const Tensor& self, const other_type& other, const Tensor& output) { \
+  TORCH_CHECK(!(self.scalar_type() == ScalarType::Long &&                                       \
+               (#func_stub == "power" || #func_stub == "atan2")),                               \
+               "MPS does not support ", #func_stub, " op with int64 input")                   \
   mps::binaryOp##other_type(self, other, Scalar(1.0), output, #func_stub,                       \
     ^BinaryOpFn(cachedGraph, primaryCastTensor, secondaryCastTensor) {                          \
       MPSGraph* mpsGraph = cachedGraph->graph();                                                \

aten/src/ATen/native/mps/operations/UnaryOps.mm

@@ -134,6 +134,7 @@ void unary_op(const Tensor& self, const Tensor& output, std::string op_name, Una
 
 TORCH_IMPL_FUNC(log1p_out_mps) (const Tensor& self, const Tensor& output)
 {
+    TORCH_CHECK(self.scalar_type() != ScalarType::Long, "MPS does not support log1p op with int64 input")
     using namespace mps;
     if (!output.is_same_size(self)) {
       output.resize_(self.sizes());