diff options
Diffstat (limited to 'src/cuda-sim/instructions.cc')
| -rw-r--r-- | src/cuda-sim/instructions.cc | 155 |
1 files changed, 151 insertions, 4 deletions
diff --git a/src/cuda-sim/instructions.cc b/src/cuda-sim/instructions.cc index aaee2a2..d0396a5 100644 --- a/src/cuda-sim/instructions.cc +++ b/src/cuda-sim/instructions.cc @@ -48,7 +48,7 @@ using half_float::half; unsigned ptx_instruction::g_num_ptx_inst_uid=0; -bool g_debug_instruction = 0; +bool g_debug_instruction = 1; const char *g_opcode_string[NUM_OPCODES] = { @@ -129,6 +129,12 @@ unsigned thread_group_offset(int thread,unsigned wmma_type,unsigned wmma_layout } } break; + case VP_MMA: + if(wmma_layout==ROW) + offset=load_c_float_row[thread_group]+16*in_tg_index; + else + offset=load_c_float_col[thread_group]+16*in_tg_index; + break; default: abort(); @@ -1709,6 +1715,14 @@ void mapping(int thread,int wmma_type,int wmma_layout,int type,int index,int str } } +void vp_mma_impl( const ptx_instruction *pI, core_t *core, warp_inst_t inst ) +{ + unsigned wmma_type = pI->get_wmma_type(); + unsigned a_layout = pI->get_wmma_layout(0); + unsigned b_layout = pI->get_wmma_layout(1); + + +} void mma_impl( const ptx_instruction *pI, core_t *core, warp_inst_t inst ) { int i,j,k,thrd; @@ -1752,7 +1766,7 @@ void mma_impl( const ptx_instruction *pI, core_t *core, warp_inst_t inst ) int hex_val; if(!((i==3)&&(type2==F32_TYPE))){ - for(k=0;k<2*nelem;k++){ + for(k=0;k<2*nelem;k++){ if(k%2==1) hex_val=(v[k/2].s64&0xffff); else @@ -2958,6 +2972,12 @@ void ldu_impl( const ptx_instruction *pI, ptx_thread_info *thread ) ld_exec(pI,thread); } +void vp_st_impl( const ptx_instruction *pI, core_t *core, warp_inst_t inst ) +{ + const operand_info &src = pI->operand_lookup(1); + const operand_info &src1 = pI->operand_lookup(0); + const operand_info &src2 = pI->operand_lookup(2); +} void mma_st_impl( const ptx_instruction *pI, core_t *core, warp_inst_t inst ) { size_t size; @@ -3000,7 +3020,7 @@ void mma_st_impl( const ptx_instruction *pI, core_t *core, warp_inst_t inst ) type_info_key::type_decode(type,size,t); if(g_debug_instruction) - printf("mma_st: thrd=%d,addr=%d, fp(size=%d), stride=%d\n",thrd,addr_reg.u32,size,src2_data.u32); + printf("mma_st: thrd=%d,addr=%x, fp(size=%d), stride=%d\n",thrd,addr_reg.u32,size,src2_data.u32); addr_t new_addr = addr+thread_group_offset(thrd,wmma_type,wmma_layout,type,stride)*size/8; ptx_reg_t nw_v[8]; @@ -3042,6 +3062,133 @@ void mma_st_impl( const ptx_instruction *pI, core_t *core, warp_inst_t inst ) thread->m_last_memory_space = space; } } +void vp_ld_impl(const ptx_instruction *pI, core_t *core, warp_inst_t inst) +{ + size_t size; + int t,i; + unsigned smid; + const operand_info &dst = pI->dst(); + const operand_info &src1 = pI->src1(); + const operand_info &src2 = pI->src2(); + unsigned type = pI->get_type(); + unsigned wmma_type = pI->get_wmma_type(); + unsigned wmma_layout = pI->get_wmma_layout(0); + int tid = inst.warp_id_func()*core->get_warp_size(); + int thrd,stride; + ptx_thread_info *thread; + for (thrd=0; thrd < core->get_warp_size(); thrd++){ + thread = core->get_thread_info()[tid+thrd]; + ptx_reg_t src1_data = thread->get_operand_value(src1, dst, U32_TYPE, thread, 1); + ptx_reg_t src2_data = thread->get_operand_value(src2, dst, U32_TYPE, thread, 1); + stride=src2_data.u32; + memory_space_t space = pI->get_space(); + + memory_space *mem = NULL; + addr_t addr = src1_data.u32; + + smid = thread->get_hw_sid(); + if( whichspace(addr) == shared_space ) { + addr= generic_to_shared(smid,addr); + space = shared_space; + } + + decode_space(space,thread,src1,mem,addr); + type_info_key::type_decode(type,size,t); + + ptx_reg_t data[8]; + addr_t new_addr; + //note we are using distribution of VP_MMA for every type of load! + if(wmma_type==LOAD_A||wmma_type==LOAD_C){ + new_addr = addr+thread_group_offset(thrd,VP_MMA,wmma_layout,type,stride)*size/8; + } + else if(wmma_type==LOAD_B4){ + new_addr = addr+thread_group_offset(thrd,VP_MMA,wmma_layout,type,stride)/8; + } + else if (wmma_type==LOAD_B8){ + new_addr = addr+thread_group_offset(thrd,VP_MMA,wmma_layout,type,stride)/4; + } + else if (wmma_type==LOAD_B16){ + new_addr = addr+thread_group_offset(thrd,VP_MMA,wmma_layout,type,stride)/2; + } + + if(g_debug_instruction) + printf("vp_ld: thrx=%d,addr=%x, base_addr=%x, size=%d, stride=%d\n",thrd,new_addr,addr,size,src2_data.u32); + + if(wmma_type==LOAD_A||wmma_type==LOAD_C){ + for(i=0;i<8;i++){ + mem->read(new_addr+4*i,size/8,&data[i].s64); + } + } + else if(wmma_type==LOAD_B4){ + mem->read(new_addr,size/8,&data[0].s64); + } + else if(wmma_type==LOAD_B8){ + mem->read(new_addr,size/8,&data[0].s64); + mem->read(new_addr+4,size/8,&data[1].s64); + } + else if(wmma_type==LOAD_B16){ + mem->read(new_addr,size/8,&data[0].s64); + mem->read(new_addr+4,size/8,&data[1].s64); + mem->read(new_addr+8,size/8,&data[2].s64); + mem->read(new_addr+12,size/8,&data[3].s64); + } + else{ + printf("wrong vp_load type\n");; + abort(); + } + + int num_reg; + if(g_debug_instruction){ + printf("\nvp_ld:thread%d= ",thrd); + if((wmma_type==LOAD_A)||(wmma_type==LOAD_C)){ + num_reg=8; + } + else if(wmma_type==LOAD_B4){ + num_reg=1; + } + else if(wmma_type==LOAD_B8){ + num_reg=2; + } + else if(wmma_type==LOAD_B16){ + num_reg=4; + } + + for(i=0;i<num_reg;i++){ + printf("%x ",data[i].s32); + } + printf("\n"); + } + + if((wmma_type==LOAD_A)||(wmma_type==LOAD_C)){ + thread->set_wmma_vector_operand_values(dst,data[0],data[1],data[2],data[3],data[4],data[5],data[6],data[7]); + } + else if(wmma_type==LOAD_B4){ + thread->set_operand_value(dst,data[0], type, thread, pI); + } + else if(wmma_type==LOAD_B8){ + thread->set_vector_operand_values(dst,data[0],data[1],data[1],data[1]); + } + else if (wmma_type==LOAD_B16){ + thread->set_vector_operand_values(dst,data[0],data[1],data[2],data[3]); + } + else + abort(); + + + + if(g_debug_instruction){ + for(int i=0;i<num_reg;i++) + printf("vp_ld:data[%i].s64=%x\n",i,data[i].u64); + } + + + thread->m_last_effective_address = addr; + thread->m_last_memory_space = space; + } + + + +} void mma_ld_impl( const ptx_instruction *pI, core_t *core, warp_inst_t inst ) { @@ -3080,7 +3227,7 @@ void mma_ld_impl( const ptx_instruction *pI, core_t *core, warp_inst_t inst ) ptx_reg_t data[16]; if(g_debug_instruction) - printf("mma_ld: thrd=%d,addr=%d, fpsize=%d, stride=%d\n",thrd,src1_data.u32,size,src2_data.u32); + printf("mma_ld: thrd=%d,addr=%x, fpsize=%d, stride=%d\n",thrd,src1_data.u32,size,src2_data.u32); addr_t new_addr = addr+thread_group_offset(thrd,wmma_type,wmma_layout,type,stride)*size/8; |