DeepSpeed icon indicating copy to clipboard operation
DeepSpeed copied to clipboard

Published 20 hours ago • verified publisher icon microsoft

[ROCm] Hip headers fix

Open rraminen opened this issue 2 years ago • 0 comments

This PR is to

  1. Hipify cg header in apply_rotary_pos_emb.cu
  2. Exclude cuda_bf16.h on ROCm

cc: @jithunnair-amd

rraminen avatar May 12 '23 17:05 rraminen