Fix the type one.

pytorch · Jun 1, 2022 · 4489ef3 · 4489ef3
1 parent cfaa917
commit 4489ef3
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/aten/src/ATen/native/mps/operations/RangeFactories.mm b/aten/src/ATen/native/mps/operations/RangeFactories.mm
@@ -103,8 +103,8 @@
             MPSGraph* mpsGraph = make_mps_graph();
             newCachedGraph = new CachedGraph(mpsGraph);
 
-            int shapeVal[1] = {(uint32_t)steps};
-            MPSGraphTensor *shapeTensor = [mpsGraph constantWithData:[NSData dataWithBytes:shapeVal length:sizeof(uint32_t)]
+            int shapeVal[1] = {(int32_t)steps};
+            MPSGraphTensor *shapeTensor = [mpsGraph constantWithData:[NSData dataWithBytes:shapeVal length:sizeof(int32_t)]
                                                                shape: @[@1]
                                                               dataType:MPSDataTypeInt32];
             MPSGraphTensor* coordsTensor = [mpsGraph coordinateAlongAxis:0
@@ -159,10 +159,7 @@
       NSDictionary<MPSGraphTensor*, MPSGraphTensorData*>* results = @{
         outputPlaceholder.getMPSGraphTensor() : outputPlaceholder.getMPSGraphTensorData()
       };
-      [cachedGraph->graph() dump];
-
       runMPSGraph(stream, cachedGraph->graph(), feeds, results);
-
     }
 
     if (!result.is_contiguous()) {