add forceinline

9472aef6 · rusty1s · 01fa24ee · 9472aef6
Commit 9472aef6 authored Apr 20, 2022 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

csrc/cuda/utils.cuh csrc/cuda/utils.cuh +3 -2

No files found.
--- a/csrc/cuda/utils.cuh
+++ b/csrc/cuda/utils.cuh
@@ -8,8 +8,9 @@
 #define CHECK_CONTIGUOUS(x)                                                    \
  AT_ASSERTM(x.is_contiguous(), #x " must be contiguous")
-__device__ int64_t get_example_idx(int64_t idx, const int64_t *ptr,
+__forceinline__ __device__ int64_t get_example_idx(int64_t idx,
-                                   const int64_t num_examples) {
+                                                   const int64_t *ptr,
+                                                   const int64_t num_examples) {
  for (int64_t i = 0; i < num_examples; i++) {
    if (ptr[i + 1] > idx)
      return i;