# ROCm 版本deepspeed 安装说明 ## OP支持情况
模块名称 环境变量 ROCm移植状态 不支持op
Utils DS_BUILD_UTILS Success --
Transformer DS_BUILD_TRANSFORMER Part Success 不支持normalize_kernels、softmax_kernels全部kernel定义、不支持general_kernels.hip中column_sum_reduce的kernel部分
StochasticTransformer DS_BUILD_STOCHASTIC_TRANSFORMER Part Success 继承的TRANSFORMER中的文件,支持列表同上
SparseAttn DS_BUILD_SPARSE_ATTN Success --
FusedLamb DS_BUILD_FUSED_LAMB Success --
FusedAdam DS_BUILD_FUSED_ADAM Success --
CPUAdam DS_BUILD_CPU_ADAM Success --
## 安装 默认安装所有op:(默认在./dist下生成编译好的whl包) ``` CXX=hipcc CC=hipcc DS_BUILD_OPS=1 python3 setup.py $VERBOSE bdist_wheel ``` 安装指定的op: ``` 指定对应op的环境变量即可,eg:DS_BUILD_FUSED_ADAM=1 ``` 安装后信息核实: ``` ds_report ```