From 5e8f4c934cfe1cd3e051e3546a1c14793f11d8d0 Mon Sep 17 00:00:00 2001
From: Philippe Tillet <phil@openai.com>
Date: Fri, 27 Nov 2020 17:00:11 -0500
Subject: [PATCH] [DRIVER] Better exception handling of invalid ptx

---
 lib/driver/module.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/driver/module.cc b/lib/driver/module.cc
index d2fe645c8..bdac2798e 100755
--- a/lib/driver/module.cc
+++ b/lib/driver/module.cc
@@ -276,7 +276,7 @@ cu_module::cu_module(std::string const & source) : module(CUmodule(), true), sou
   void* optval[] = {(void*)(uintptr_t)errbufsize, (void*)errbuf.data()};
   try{
     dispatch::cuModuleLoadDataEx(&*cu_, source_.data(), 2, opt, optval);
-  }catch(exception::cuda::base const &){
+  }catch(exception::cuda::invalid_ptx const &){
 //#ifdef TRITON_LOG_PTX_ERROR
     std::cout << source << std::endl;
     std::cerr << "It appears that Triton produced invalid PTX code:" << std::endl;