diff --git a/faiss/gpu/GpuIndex.cu b/faiss/gpu/GpuIndex.cu index d667ae149..d1ae3b538 100644 --- a/faiss/gpu/GpuIndex.cu +++ b/faiss/gpu/GpuIndex.cu @@ -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 } };