add get func (PaddlePaddle#60)

zmxdream · Sep 14, 2022 · a27936c · a27936c
1 parent 9b1a063
commit a27936c
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/paddle/fluid/framework/fleet/heter_ps/feature_value.h b/paddle/fluid/framework/fleet/heter_ps/feature_value.h
@@ -301,6 +301,23 @@ class CommonFeatureValueAccessor {
   __host__ __device__ CommonFeatureValueAccessor() {}
   __host__ __device__ ~CommonFeatureValueAccessor() {}
 
+
+  #define DEFINE_GET_INDEX4( instance, field) \
+        __host__ __device__ int get_##field##_index() { \
+            return instance.field##Index(); \
+        }
+
+#ifdef PADDLE_WITH_PSLIB
+  DEFINE_GET_INDEX4(common_feature_value, Show)
+  DEFINE_GET_INDEX4(common_feature_value, Click)
+  DEFINE_GET_INDEX4(common_feature_value, EmbedW)
+  DEFINE_GET_INDEX4(common_feature_value, EmbedxW)
+  DEFINE_GET_INDEX4(common_feature_value, DeltaScore)
+  DEFINE_GET_INDEX4(common_feature_value, Slot)
+  DEFINE_GET_INDEX4(common_feature_value, EmbedG2Sum)
+  DEFINE_GET_INDEX4(common_feature_value, MfDim)
+#endif
+
   __host__ int Initialize() {
     // NOTE(zhangminxu): gpups' sparse table optimizer type,
     // now only support embed&embedx 's sparse optimizer is the same