• Masaki Kozuki's avatar
    remove THC headers/functions (#1192) · 2155dabf
    Masaki Kozuki authored
    Changes include
    - THC headers removal
    - TH macros replacement
    - fix some typo in comment
     Conflicts:
    	apex/contrib/csrc/multihead_attn/additive_masked_softmax_dropout_cuda.cu
    	apex/contrib/csrc/multihead_attn/encdec_multihead_attn_cuda.cu
    	apex/contrib/csrc/multihead_attn/encdec_multihead_attn_norm_add_cuda.cu
    	apex/contrib/csrc/multihead_attn/masked_softmax_dropout_cuda.cu
    	apex/contrib/csrc/multihead_attn/self_multihead_attn_bias_additive_mask_cuda.cu
    	apex/contrib/csrc/multihead_attn/self_multihead_attn_bias_cuda.cu
    	apex/contrib/csrc/multihead_attn/self_multihead_attn_cuda.cu
    	apex/contrib/csrc/multihead_attn/self_multihead_attn_norm_add_cuda.cu
    	apex/contrib/csrc/multihead_attn/strided_batched_gemm.h
    2155dabf
batch_norm.cu 11.6 KB