From 29a0ad6c4dcda831c4df08a0b09c4e997298b716 Mon Sep 17 00:00:00 2001 From: Philippe Tillet Date: Wed, 29 Jul 2020 13:50:01 -0400 Subject: [PATCH] [DRIVER] Now always using PTXv6.4 --- lib/driver/module.cc | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/lib/driver/module.cc b/lib/driver/module.cc index 1af681518..99a1e5405 100755 --- a/lib/driver/module.cc +++ b/lib/driver/module.cc @@ -251,11 +251,10 @@ std::string cu_module::compile_llvm_module(std::unique_ptr module, int version; dispatch::cuDriverGetVersion(&version); int major = version / 1000; - int minor = (version - major*1000) / 10; +// int minor = (version - major*1000) / 10; if(major < 10) throw std::runtime_error("Triton requires CUDA 10+"); - if(minor >= 1) - find_and_replace(result, ".version", "\n", ".version 6.4\n"); + find_and_replace(result, ".version", "\n", ".version 6.4\n"); while(find_and_replace(result, "\t// begin inline asm", "\n", "")); while(find_and_replace(result, "\t// end inline asm", "\n", "")); return result;