• J
    Added mul BF16/FP32 FWD/BWD oneDNN kernel (#38552) · fc6eed5b
    jakpiase 提交于
    * base changes for mul reimplementation
    
    * empty commit
    
    * tmp save
    
    * full implementation of mul bf16/fp32 fwd bwd
    
    * CI fix
    
    * CI rerun
    
    * changed unity build cmake to avoid gpu issues
    
    * removed mul mkldnn from unity build
    
    * added skipping tests if not cpu_bf16
    
    * CI fix
    
    * CI fix
    
    * CI fix
    fc6eed5b
matmul_v2_mkldnn_op.cc 16.4 KB