diff --git a/fbgemm_gpu/src/sparse_ops/common.cuh b/fbgemm_gpu/src/sparse_ops/common.cuh
index c1ce3a8709..a755ddc022 100644
--- a/fbgemm_gpu/src/sparse_ops/common.cuh
+++ b/fbgemm_gpu/src/sparse_ops/common.cuh
@@ -20,6 +20,7 @@
 #include <c10/cuda/CUDADeviceAssertionHost.h>
 #include <c10/cuda/CUDAGuard.h>
 #include <torch/library.h>
+#include <type_traits>
 
 // clang-format off
 #include "fbgemm_gpu/utils/cub_namespace_prefix.cuh"