Fix FBGEMM amd build (pytorch#2189)

Summary: Pull Request resolved: pytorch#2189 Reviewed By: q10 Differential Revision: D51609251 fbshipit-source-id: d6ae1a74ebfa9b207b271866a157f73a0b562298
ROCm · Dec 10, 2023 · 0026ddc · 0026ddc
1 parent 8724d89
commit 0026ddc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fbgemm_gpu/src/sparse_ops/common.cuh b/fbgemm_gpu/src/sparse_ops/common.cuh
@@ -32,7 +32,7 @@
 #include "fbgemm_gpu/split_embeddings_utils.cuh"
 
 #ifdef USE_ROCM
-#include <hipblas.h>
+#include <hipblas/hipblas.h>
 #endif
 
 #ifdef USE_ROCM