9520f251808add32cd7c6b36f65fc6e031eaa3ff,nni/algorithms/compression/pytorch/quantization/quantizers.py,,update_quantization_param,#Any#Any#Any#,68

Before Change

    // representable value.
    if rmin.is_cuda:
        rmin = torch.min(rmin, torch.Tensor([0]).cuda())
        rmax = torch.max(rmax, torch.Tensor([0]).cuda())
        qmin = torch.Tensor([0]).cuda()
        qmax = torch.Tensor([(1 << bits) - 1]).cuda()
    else:
        rmin = torch.min(rmin, torch.Tensor([0]))

After Change

    // extend the [min, max] interval to ensure that it contains 0.
    // Otherwise, we would not meet the requirement that 0 be an exactly
    // representable value.
    rmin = torch.min(rmin, torch.Tensor([0]).to(rmin.device))
    rmax = torch.max(rmax, torch.Tensor([0]).to(rmin.device))
    qmin = torch.Tensor([0]).to(rmin.device)
    qmax = torch.Tensor([(1 << bits) - 1]).to(rmin.device)

Instances