add AMD_ROCM as part of get_compile_options (#3790)

Summary:
Pull Request resolved: https://github.com/facebookresearch/faiss/pull/3790

Comment from ramilbakhshyiev https://www.internalfb.com/diff/D61674490?dst_version_fbid=359297527230868&transaction_fbid=8025001410919172

Reviewed By: ramilbakhshyiev

Differential Revision: D61680186

fbshipit-source-id: 2b6d5803e620b36878b669e617253c875562c30f
pull/3798/head
Mengdi Lin 2024-08-23 11:08:39 -07:00 committed by Facebook GitHub Bot
parent 58a673d938
commit 4ca67340ea
1 changed files with 4 additions and 0 deletions

View File

@ -537,6 +537,10 @@ struct InitGpuCompileOptions {
#ifdef USE_NVIDIA_RAFT
gpu_compile_options += "NVIDIA_RAFT ";
#endif
#ifdef USE_AMD_ROCM
gpu_compile_options += "AMD_ROCM ";
#endif
}
};