Skip to content

Commit ee8aea9

Browse files
committed
fix cuda build
1 parent 9cd52cf commit ee8aea9

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

faiss/gpu/impl/PQCodeLoad.cuh

+3-3
Original file line numberDiff line numberDiff line change
@@ -47,9 +47,6 @@ inline __device__ unsigned int getByte(uint64_t v, int pos, int width) {
4747
return getBitfield(v, pos, width);
4848
}
4949

50-
// template <int NumSubQuantizers>
51-
// struct LoadCode32 {};
52-
5350
#ifdef USE_ROCM
5451

5552
template <int NumSubQuantizers>
@@ -381,6 +378,9 @@ struct LoadCode32<96> {
381378

382379
#else // USE_ROCM
383380

381+
template <int NumSubQuantizers>
382+
struct LoadCode32 {};
383+
384384
template <>
385385
struct LoadCode32<1> {
386386
static inline __device__ void load(

faiss/gpu/utils/PtxUtils.cuh

+2
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,9 @@
88
#pragma once
99

1010
#include <cuda.h>
11+
#ifdef USE_ROCM
1112
#include <device_functions.h>
13+
#endif
1214

1315
namespace faiss {
1416
namespace gpu {

0 commit comments

Comments
 (0)