update seed and add TODO

PaddlePaddle · Sep 2, 2022 · 400381c · 400381c
1 parent 026188d
commit 400381c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/paddlenlp/ops/faster_transformer/src/CMakeLists.txt b/paddlenlp/ops/faster_transformer/src/CMakeLists.txt
@@ -225,6 +225,8 @@ else(ON_INFER)
                 RESULT_VARIABLE _INC_PYTHON_SUCCESS
                 OUTPUT_VARIABLE _INC_PYTHON_VALUES)
   message(STATUS "PADDLE_VERSION: ${_INC_PYTHON_VALUES}")
+
+  # TODO(gongenlei): support PADDLE_NEW_ALLOCATOR for ON_INFER
   if(_INC_PYTHON_VALUES VERSION_GREATER_EQUAL "2.3.0")
       add_definitions(-DPADDLE_NEW_ALLOCATOR)
   endif()

diff --git a/paddlenlp/ops/patches/FasterTransformer/fastertransformer/cuda/topk_kernels.cu b/paddlenlp/ops/patches/FasterTransformer/fastertransformer/cuda/topk_kernels.cu
@@ -41,7 +41,7 @@ __global__ void ker_curand_setup_bsz_one(curandState_t* state,
   if (threadIdx.x + blockIdx.x * blockDim.x < size)
     curand_init(seed,
                 0,
-                0,
+                seed,
                 &state[blockIdx.x * blockDim.x + threadIdx.x]);
 }