message: [XLA:GPU] Warn if using driver jit < 396.20.0.
  396.20 is the first driver version that contains ptxas 9.2.88, which is
  the oldest known-working ptxas at the moment.
diff: @@ -362,8 +362,8 @@ void WarnIfBadPtxasVersion(const string& ptxas_path) {
  // b/70245379.
  //
  // ptxas 9.1.121 miscompiles some large multioutput fusions, again in a way
  - // that appears related to address calculations. ptxas 9.2.88 appears to
  - // work, as far as we can tell.
  + // that appears related to address calculations, b/111107644. ptxas 9.2.88
  + // appears to work, as far as we can tell.
  if (vmaj < 9) {
  LOG(ERROR)
  << "You are using ptxas 8.x, but XLA requires ptxas 9.x (and strongly "
  @@ -406,20 +406,17 @@ void WarnIfBadDriverJITVersion() {
  // - 387.x before 387.40
  // - 390.x before 390.10.
  //
  - // TODO(jlebar): This list does not cover the address-calculation bug we've
  - // observed in ptxas 9.1.121. Need to get a new safe range from nvidia
  - // corresponding to ptxas >= 9.2.88.
  - auto vmaj = std::get<0>(version);
  - auto vmin = std::get<1>(version);
  - if ((vmaj == 384 && vmin < 108) || //
  - (vmaj == 387 && vmin < 40) || //
  - (vmaj == 390 && vmin < 10)) {
  + // In addition, only >= 396.20 contains ptxas >= 9.2.88, which contains the
  + // fix for the "large multioutput fusions" miscompile, b/111107644.
  + if (version < std::make_tuple(396, 20, 0)) {
  LOG(WARNING)
  << "*** WARNING *** Invoking the PTX->SASS JIT from driver version "
  << se::cuda::DriverVersionToString(version)
  - << ", which is in range [384.0.0, 384.108.0) + [387.0.0, 387.40.0) + "
  - "[390.0.0, 390.10.0). These versions are known to miscompile XLA "
  - "code, leading to incorrect results or invalid-address errors.";
  + << ", which is older than 396.20.0. These versions are known to "
  + "miscompile XLA code, leading to incorrect results or "
  + "invalid-address errors.\nXLA only uses the driver JIT if it "
  + "cannot find ptxas; you don't need to update your driver if "
  + "you can point XLA to ptxas 9.2.88 or newer.";
  }
  });
  }
  