remove cast to long. apache/tvm#6311 fixed it

microsoft · Aug 20, 2020 · ca5b854 · ca5b854
1 parent 5aee31a
commit ca5b854
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/hummingbird/ml/operator_converters/_tree_implementations.py b/hummingbird/ml/operator_converters/_tree_implementations.py
@@ -220,7 +220,6 @@ def aggregation(self, x):
         return x
 
     def forward(self, x):
-        self.nodes_offset.long()  # This is necessary otherwise TVM does not compile. See https://github.com/apache/incubator-tvm/issues/6300
         indexes = self.nodes_offset
         indexes = indexes.expand(x.size()[0], self.num_trees)
         indexes = indexes.reshape(-1)
@@ -318,7 +317,6 @@ def aggregation(self, x):
         return x
 
     def forward(self, x):
-        self.tree_indices.long()  # This is necessary otherwise TVM does not compile. See https://github.com/apache/incubator-tvm/issues/6300
         prev_indices = (torch.ge(torch.index_select(x, 1, self.root_nodes), self.root_biases)).long()
         prev_indices = prev_indices + self.tree_indices
         prev_indices = prev_indices.view(-1)