db0215eeec9ce23e7cc75577828755095c2319d5,python/tvm/relay/frontend/pytorch.py,PyTorchOpConverter,bincount,#PyTorchOpConverter#Any#Any#,1921

Before Change

        maximum = _op.max(data)
        dim = maximum + _expr.const(1, dtype="int32")
        if weights:
            weight_type = _infer_type(weights).checked_type
            out_dtype = weight_type.dtype
            updates = weights
        else:
            out_dtype = "int32"

After Change

    def bincount(self, inputs, input_types):
        data = inputs[0]
        weights = inputs[1]
        input_type = self.infer_type(data).dtype
        if input_type == "int64":
            logging.warning(
                "Casting an int64 input to int32, since we do not have int64 atomic add"

Instances