diff --git a/mlir/lib/Dialect/GPU/Transforms/SerializeToCubin.cpp b/mlir/lib/Dialect/GPU/Transforms/SerializeToCubin.cpp --- a/mlir/lib/Dialect/GPU/Transforms/SerializeToCubin.cpp +++ b/mlir/lib/Dialect/GPU/Transforms/SerializeToCubin.cpp @@ -160,7 +160,7 @@ MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(SerializeToCubinPass) SerializeToCubinPass(StringRef triple = "nvptx64-nvidia-cuda", - StringRef chip = "sm_35", StringRef features = "+ptx60", + StringRef chip = "sm_50", StringRef features = "+ptx60", int optLevel = 2, bool dumpPtx = false, bool usePtxas = true, StringRef ptxasParams = {}); diff --git a/mlir/test/lib/Dialect/GPU/TestConvertGPUKernelToCubin.cpp b/mlir/test/lib/Dialect/GPU/TestConvertGPUKernelToCubin.cpp --- a/mlir/test/lib/Dialect/GPU/TestConvertGPUKernelToCubin.cpp +++ b/mlir/test/lib/Dialect/GPU/TestConvertGPUKernelToCubin.cpp @@ -38,7 +38,7 @@ TestSerializeToCubinPass::TestSerializeToCubinPass() { this->triple = "nvptx64-nvidia-cuda"; - this->chip = "sm_35"; + this->chip = "sm_50"; this->features = "+ptx60"; }