| 2923 | } |
| 2924 | |
| 2925 | void SlotRecordInMemoryDataFeed::DumpSampleNeighbors(std::string dump_path) { |
| 2926 | VLOG(1) << "INTO SlotRecordInMemoryDataFeed::DumpSampleNeighbors"; |
| 2927 | #if defined(PADDLE_WITH_PSCORE) && defined(PADDLE_WITH_HETERPS) |
| 2928 | if (gpu_graph_mode_) { |
| 2929 | std::string path = |
| 2930 | string::format_string("%s/part-%03d", dump_path.c_str(), thread_id_); |
| 2931 | gpu_graph_data_generator_.DumpSampleNeighbors(path); |
| 2932 | } |
| 2933 | #endif |
| 2934 | } |
| 2935 | |
| 2936 | #if defined(PADDLE_WITH_CUDA) && defined(PADDLE_WITH_HETERPS) |
| 2937 | void SlotRecordInMemoryDataFeed::BuildSlotBatchGPU(const int ins_num, |
nothing calls this directly
no test coverage detected