1 files changed, 18 insertions, 0 deletions
diff --git a/src/cuda-sim/ptx_ir.cc b/src/cuda-sim/ptx_ir.cc
index 79dc2b0..d5e8df2 100644
--- a/src/cuda-sim/ptx_ir.cc
+++ b/src/cuda-sim/ptx_ir.cc
@@ -731,6 +731,18 @@ void symbol_table::set_name( const char *name )
    m_scope_name = std::string(name);
 }
 
+const ptx_version &symbol_table::get_ptx_version() const 
+{ 
+   if( m_parent == NULL ) return m_ptx_version;
+   else return m_parent->get_ptx_version(); 
+}
+
+void symbol_table::set_ptx_version( float ver, unsigned ext ) 
+{ 
+   m_ptx_version = ptx_version(ver,ext); 
+   assert( m_ptx_version.ver() < 3 );
+}
+
 symbol *symbol_table::lookup( const char *identifier ) 
 {
    std::string key(identifier);
@@ -1241,6 +1253,12 @@ void function_info::print_basic_block_dot()
    printf("}\n");
 }
 
+extern "C" void add_version_info( float ver )
+{
+   g_global_symbol_table->set_ptx_version(ver,0);
+}
+
+
 extern "C" void add_file( unsigned num, const char *filename )
 {
    if( g_filename == NULL ) {