format

qingshui · Feb 23, 2024 · a4297df · a4297df
1 parent bb31426
commit a4297df
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/paddle/fluid/operators/fused/fused_seqpool_cvm_op.cu b/paddle/fluid/operators/fused/fused_seqpool_cvm_op.cu
@@ -339,15 +339,14 @@ __global__ void FusedCVMKernelWithShowConcate(const size_t N,
     int x = key / batch_size;  // slot id
     int y = key % batch_size;  // ins id
     const T *in =
-        &seqpool_output_values[key * embedding_size * embedx_concate_size];
-
+        &seqpool_output_values[key * embedding_size * embedx_concate_size +
+                               k * embedding_size];
+    T *out = (output_values[x] + y * concat_embedding_size +
+              k * noclk_embedding_size + offset);
     if (offset == 0) {  // show
-      *(output_values[x] + y * concat_embedding_size +
-        k * noclk_embedding_size) = log(in[k * embedding_size] + 1);
+      *out = log(in[0] + 1);
     } else {  // skip click offset + 1
-      *(output_values[x] + y * concat_embedding_size +
-        k * noclk_embedding_size + offset) =
-          in[k * embedding_size + offset + 1];
+      *out = in[offset + 1];
     }
   }
 }
@@ -394,11 +393,12 @@ __global__ void FusedCVMKernelNoCVMEmbedxConcate(
     int x = key / batch_size;  // slot id
     int y = key % batch_size;  // ins id
     const T *in =
-        &seqpool_output_values[key * embedding_size * embedx_concate_size];
+        &seqpool_output_values[key * embedding_size * embedx_concate_size +
+                               k * embedding_size];
 
     // no cvm
     *(output_values[x] + y * concat_embedding_size + k * no_cvm_embedding_size +
-      offset) = in[k * embedding_size + offset + cvm_offset];
+      offset) = in[offset + cvm_offset];
   }
 }