diff options
| author | aamir <[email protected]> | 2018-06-05 12:50:57 -0700 |
|---|---|---|
| committer | aamir <[email protected]> | 2018-06-05 12:50:57 -0700 |
| commit | 4161ccba0d4a99157afed3cdccef0e9c2a6d89e6 (patch) | |
| tree | 9752d35aa66ff33923a5be8243dce2522c37029a /src/cuda-sim/ptx_ir.cc | |
| parent | 708031c0274a730dfd99820fd49351785a60e2d7 (diff) | |
added support for wmma:load_c:f16_type
Diffstat (limited to 'src/cuda-sim/ptx_ir.cc')
| -rw-r--r-- | src/cuda-sim/ptx_ir.cc | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/src/cuda-sim/ptx_ir.cc b/src/cuda-sim/ptx_ir.cc index 9a4d8d3..fb9adca 100644 --- a/src/cuda-sim/ptx_ir.cc +++ b/src/cuda-sim/ptx_ir.cc @@ -1083,6 +1083,27 @@ ptx_instruction::ptx_instruction( int opcode, int rr=0; std::list<int>::const_iterator i; unsigned n=1; + for ( i=wmma_options.begin(); i!= wmma_options.end(); i++, n++ ) { + int last_ptx_inst_option = *i; + switch ( last_ptx_inst_option ) { + case SYNC_OPTION: + case LOAD_A: + case LOAD_B: + case LOAD_C: + case STORE_D: + case MMA: + m_wmma_type=last_ptx_inst_option; + break; + case ROW: + case COL: + case M16N16K16: + break; + default: + assert(0); + break; + } + } + n=1; for ( i=options.begin(); i!= options.end(); i++, n++ ) { int last_ptx_inst_option = *i; switch ( last_ptx_inst_option ) { |
