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: 2b6d5803e620b36878b669e617253c875562c30fpull/3798/head
parent
58a673d938
commit
4ca67340ea
|
@ -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
|
||||
}
|
||||
};
|
||||
|
||||
|
|
Loading…
Reference in New Issue