Commit acf1dbd4 by XXX

update v11: Synchronize multithreaded communication modifications

parent eb2cca25
set(CMAKE_C_COMPILER "/usr/bin/cc")
set(CMAKE_C_COMPILER_ARG1 "")
set(CMAKE_C_COMPILER_ID "GNU")
set(CMAKE_C_COMPILER_VERSION "9.4.0")
set(CMAKE_C_COMPILER_VERSION_INTERNAL "")
set(CMAKE_C_COMPILER_WRAPPER "")
set(CMAKE_C_STANDARD_COMPUTED_DEFAULT "11")
set(CMAKE_C_COMPILE_FEATURES "c_std_90;c_function_prototypes;c_std_99;c_restrict;c_variadic_macros;c_std_11;c_static_assert")
set(CMAKE_C90_COMPILE_FEATURES "c_std_90;c_function_prototypes")
set(CMAKE_C99_COMPILE_FEATURES "c_std_99;c_restrict;c_variadic_macros")
set(CMAKE_C11_COMPILE_FEATURES "c_std_11;c_static_assert")
set(CMAKE_C_PLATFORM_ID "Linux")
set(CMAKE_C_SIMULATE_ID "")
set(CMAKE_C_COMPILER_FRONTEND_VARIANT "")
set(CMAKE_C_SIMULATE_VERSION "")
set(CMAKE_AR "/usr/bin/ar")
set(CMAKE_C_COMPILER_AR "/usr/bin/gcc-ar-9")
set(CMAKE_RANLIB "/usr/bin/ranlib")
set(CMAKE_C_COMPILER_RANLIB "/usr/bin/gcc-ranlib-9")
set(CMAKE_LINKER "/usr/bin/ld")
set(CMAKE_MT "")
set(CMAKE_COMPILER_IS_GNUCC 1)
set(CMAKE_C_COMPILER_LOADED 1)
set(CMAKE_C_COMPILER_WORKS TRUE)
set(CMAKE_C_ABI_COMPILED TRUE)
set(CMAKE_COMPILER_IS_MINGW )
set(CMAKE_COMPILER_IS_CYGWIN )
if(CMAKE_COMPILER_IS_CYGWIN)
set(CYGWIN 1)
set(UNIX 1)
endif()
set(CMAKE_C_COMPILER_ENV_VAR "CC")
if(CMAKE_COMPILER_IS_MINGW)
set(MINGW 1)
endif()
set(CMAKE_C_COMPILER_ID_RUN 1)
set(CMAKE_C_SOURCE_FILE_EXTENSIONS c;m)
set(CMAKE_C_IGNORE_EXTENSIONS h;H;o;O;obj;OBJ;def;DEF;rc;RC)
set(CMAKE_C_LINKER_PREFERENCE 10)
# Save compiler ABI information.
set(CMAKE_C_SIZEOF_DATA_PTR "8")
set(CMAKE_C_COMPILER_ABI "ELF")
set(CMAKE_C_LIBRARY_ARCHITECTURE "x86_64-linux-gnu")
if(CMAKE_C_SIZEOF_DATA_PTR)
set(CMAKE_SIZEOF_VOID_P "${CMAKE_C_SIZEOF_DATA_PTR}")
endif()
if(CMAKE_C_COMPILER_ABI)
set(CMAKE_INTERNAL_PLATFORM_ABI "${CMAKE_C_COMPILER_ABI}")
endif()
if(CMAKE_C_LIBRARY_ARCHITECTURE)
set(CMAKE_LIBRARY_ARCHITECTURE "x86_64-linux-gnu")
endif()
set(CMAKE_C_CL_SHOWINCLUDES_PREFIX "")
if(CMAKE_C_CL_SHOWINCLUDES_PREFIX)
set(CMAKE_CL_SHOWINCLUDES_PREFIX "${CMAKE_C_CL_SHOWINCLUDES_PREFIX}")
endif()
set(CMAKE_C_IMPLICIT_INCLUDE_DIRECTORIES "/usr/lib/gcc/x86_64-linux-gnu/9/include;/usr/local/include;/usr/include/x86_64-linux-gnu;/usr/include")
set(CMAKE_C_IMPLICIT_LINK_LIBRARIES "gcc;gcc_s;c;gcc;gcc_s")
set(CMAKE_C_IMPLICIT_LINK_DIRECTORIES "/usr/lib/gcc/x86_64-linux-gnu/9;/usr/lib/x86_64-linux-gnu;/usr/lib;/lib/x86_64-linux-gnu;/lib")
set(CMAKE_C_IMPLICIT_LINK_FRAMEWORK_DIRECTORIES "")
set(CMAKE_CXX_COMPILER "/usr/bin/c++")
set(CMAKE_CXX_COMPILER_ARG1 "")
set(CMAKE_CXX_COMPILER_ID "GNU")
set(CMAKE_CXX_COMPILER_VERSION "9.4.0")
set(CMAKE_CXX_COMPILER_VERSION_INTERNAL "")
set(CMAKE_CXX_COMPILER_WRAPPER "")
set(CMAKE_CXX_STANDARD_COMPUTED_DEFAULT "14")
set(CMAKE_CXX_COMPILE_FEATURES "cxx_std_98;cxx_template_template_parameters;cxx_std_11;cxx_alias_templates;cxx_alignas;cxx_alignof;cxx_attributes;cxx_auto_type;cxx_constexpr;cxx_decltype;cxx_decltype_incomplete_return_types;cxx_default_function_template_args;cxx_defaulted_functions;cxx_defaulted_move_initializers;cxx_delegating_constructors;cxx_deleted_functions;cxx_enum_forward_declarations;cxx_explicit_conversions;cxx_extended_friend_declarations;cxx_extern_templates;cxx_final;cxx_func_identifier;cxx_generalized_initializers;cxx_inheriting_constructors;cxx_inline_namespaces;cxx_lambdas;cxx_local_type_template_args;cxx_long_long_type;cxx_noexcept;cxx_nonstatic_member_init;cxx_nullptr;cxx_override;cxx_range_for;cxx_raw_string_literals;cxx_reference_qualified_functions;cxx_right_angle_brackets;cxx_rvalue_references;cxx_sizeof_member;cxx_static_assert;cxx_strong_enums;cxx_thread_local;cxx_trailing_return_types;cxx_unicode_literals;cxx_uniform_initialization;cxx_unrestricted_unions;cxx_user_literals;cxx_variadic_macros;cxx_variadic_templates;cxx_std_14;cxx_aggregate_default_initializers;cxx_attribute_deprecated;cxx_binary_literals;cxx_contextual_conversions;cxx_decltype_auto;cxx_digit_separators;cxx_generic_lambdas;cxx_lambda_init_captures;cxx_relaxed_constexpr;cxx_return_type_deduction;cxx_variable_templates;cxx_std_17;cxx_std_20")
set(CMAKE_CXX98_COMPILE_FEATURES "cxx_std_98;cxx_template_template_parameters")
set(CMAKE_CXX11_COMPILE_FEATURES "cxx_std_11;cxx_alias_templates;cxx_alignas;cxx_alignof;cxx_attributes;cxx_auto_type;cxx_constexpr;cxx_decltype;cxx_decltype_incomplete_return_types;cxx_default_function_template_args;cxx_defaulted_functions;cxx_defaulted_move_initializers;cxx_delegating_constructors;cxx_deleted_functions;cxx_enum_forward_declarations;cxx_explicit_conversions;cxx_extended_friend_declarations;cxx_extern_templates;cxx_final;cxx_func_identifier;cxx_generalized_initializers;cxx_inheriting_constructors;cxx_inline_namespaces;cxx_lambdas;cxx_local_type_template_args;cxx_long_long_type;cxx_noexcept;cxx_nonstatic_member_init;cxx_nullptr;cxx_override;cxx_range_for;cxx_raw_string_literals;cxx_reference_qualified_functions;cxx_right_angle_brackets;cxx_rvalue_references;cxx_sizeof_member;cxx_static_assert;cxx_strong_enums;cxx_thread_local;cxx_trailing_return_types;cxx_unicode_literals;cxx_uniform_initialization;cxx_unrestricted_unions;cxx_user_literals;cxx_variadic_macros;cxx_variadic_templates")
set(CMAKE_CXX14_COMPILE_FEATURES "cxx_std_14;cxx_aggregate_default_initializers;cxx_attribute_deprecated;cxx_binary_literals;cxx_contextual_conversions;cxx_decltype_auto;cxx_digit_separators;cxx_generic_lambdas;cxx_lambda_init_captures;cxx_relaxed_constexpr;cxx_return_type_deduction;cxx_variable_templates")
set(CMAKE_CXX17_COMPILE_FEATURES "cxx_std_17")
set(CMAKE_CXX20_COMPILE_FEATURES "cxx_std_20")
set(CMAKE_CXX_PLATFORM_ID "Linux")
set(CMAKE_CXX_SIMULATE_ID "")
set(CMAKE_CXX_COMPILER_FRONTEND_VARIANT "")
set(CMAKE_CXX_SIMULATE_VERSION "")
set(CMAKE_AR "/usr/bin/ar")
set(CMAKE_CXX_COMPILER_AR "/usr/bin/gcc-ar-9")
set(CMAKE_RANLIB "/usr/bin/ranlib")
set(CMAKE_CXX_COMPILER_RANLIB "/usr/bin/gcc-ranlib-9")
set(CMAKE_LINKER "/usr/bin/ld")
set(CMAKE_MT "")
set(CMAKE_COMPILER_IS_GNUCXX 1)
set(CMAKE_CXX_COMPILER_LOADED 1)
set(CMAKE_CXX_COMPILER_WORKS TRUE)
set(CMAKE_CXX_ABI_COMPILED TRUE)
set(CMAKE_COMPILER_IS_MINGW )
set(CMAKE_COMPILER_IS_CYGWIN )
if(CMAKE_COMPILER_IS_CYGWIN)
set(CYGWIN 1)
set(UNIX 1)
endif()
set(CMAKE_CXX_COMPILER_ENV_VAR "CXX")
if(CMAKE_COMPILER_IS_MINGW)
set(MINGW 1)
endif()
set(CMAKE_CXX_COMPILER_ID_RUN 1)
set(CMAKE_CXX_SOURCE_FILE_EXTENSIONS C;M;c++;cc;cpp;cxx;m;mm;CPP)
set(CMAKE_CXX_IGNORE_EXTENSIONS inl;h;hpp;HPP;H;o;O;obj;OBJ;def;DEF;rc;RC)
foreach (lang C OBJC OBJCXX)
if (CMAKE_${lang}_COMPILER_ID_RUN)
foreach(extension IN LISTS CMAKE_${lang}_SOURCE_FILE_EXTENSIONS)
list(REMOVE_ITEM CMAKE_CXX_SOURCE_FILE_EXTENSIONS ${extension})
endforeach()
endif()
endforeach()
set(CMAKE_CXX_LINKER_PREFERENCE 30)
set(CMAKE_CXX_LINKER_PREFERENCE_PROPAGATES 1)
# Save compiler ABI information.
set(CMAKE_CXX_SIZEOF_DATA_PTR "8")
set(CMAKE_CXX_COMPILER_ABI "ELF")
set(CMAKE_CXX_LIBRARY_ARCHITECTURE "x86_64-linux-gnu")
if(CMAKE_CXX_SIZEOF_DATA_PTR)
set(CMAKE_SIZEOF_VOID_P "${CMAKE_CXX_SIZEOF_DATA_PTR}")
endif()
if(CMAKE_CXX_COMPILER_ABI)
set(CMAKE_INTERNAL_PLATFORM_ABI "${CMAKE_CXX_COMPILER_ABI}")
endif()
if(CMAKE_CXX_LIBRARY_ARCHITECTURE)
set(CMAKE_LIBRARY_ARCHITECTURE "x86_64-linux-gnu")
endif()
set(CMAKE_CXX_CL_SHOWINCLUDES_PREFIX "")
if(CMAKE_CXX_CL_SHOWINCLUDES_PREFIX)
set(CMAKE_CL_SHOWINCLUDES_PREFIX "${CMAKE_CXX_CL_SHOWINCLUDES_PREFIX}")
endif()
set(CMAKE_CXX_IMPLICIT_INCLUDE_DIRECTORIES "/usr/include/c++/9;/usr/include/x86_64-linux-gnu/c++/9;/usr/include/c++/9/backward;/usr/lib/gcc/x86_64-linux-gnu/9/include;/usr/local/include;/usr/include/x86_64-linux-gnu;/usr/include")
set(CMAKE_CXX_IMPLICIT_LINK_LIBRARIES "stdc++;m;gcc_s;gcc;c;gcc_s;gcc")
set(CMAKE_CXX_IMPLICIT_LINK_DIRECTORIES "/usr/lib/gcc/x86_64-linux-gnu/9;/usr/lib/x86_64-linux-gnu;/usr/lib;/lib/x86_64-linux-gnu;/lib")
set(CMAKE_CXX_IMPLICIT_LINK_FRAMEWORK_DIRECTORIES "")
set(CMAKE_HOST_SYSTEM "Linux-5.4.0-144-generic")
set(CMAKE_HOST_SYSTEM_NAME "Linux")
set(CMAKE_HOST_SYSTEM_VERSION "5.4.0-144-generic")
set(CMAKE_HOST_SYSTEM_PROCESSOR "x86_64")
set(CMAKE_SYSTEM "Linux-5.4.0-144-generic")
set(CMAKE_SYSTEM_NAME "Linux")
set(CMAKE_SYSTEM_VERSION "5.4.0-144-generic")
set(CMAKE_SYSTEM_PROCESSOR "x86_64")
set(CMAKE_CROSSCOMPILING "FALSE")
set(CMAKE_SYSTEM_LOADED 1)
# CMAKE generated file: DO NOT EDIT!
# Generated by "Unix Makefiles" Generator, CMake Version 3.16
# Relative path conversion top directories.
set(CMAKE_RELATIVE_PATH_TOP_SOURCE "/home/sxx/zlj/rpc_ps/Sample")
set(CMAKE_RELATIVE_PATH_TOP_BINARY "/home/sxx/zlj/rpc_ps/Sample")
# Force unix paths in dependencies.
set(CMAKE_FORCE_UNIX_PATHS 1)
# The C and CXX include file regular expressions for this directory.
set(CMAKE_C_INCLUDE_REGEX_SCAN "^.*$")
set(CMAKE_C_INCLUDE_REGEX_COMPLAIN "^$")
set(CMAKE_CXX_INCLUDE_REGEX_SCAN ${CMAKE_C_INCLUDE_REGEX_SCAN})
set(CMAKE_CXX_INCLUDE_REGEX_COMPLAIN ${CMAKE_C_INCLUDE_REGEX_COMPLAIN})
# CMAKE generated file: DO NOT EDIT!
# Generated by "Unix Makefiles" Generator, CMake Version 3.16
# The generator used is:
set(CMAKE_DEPENDS_GENERATOR "Unix Makefiles")
# The top level Makefile was generated from the following files:
set(CMAKE_MAKEFILE_DEPENDS
"CMakeCache.txt"
"CMakeFiles/3.16.3/CMakeCCompiler.cmake"
"CMakeFiles/3.16.3/CMakeCXXCompiler.cmake"
"CMakeFiles/3.16.3/CMakeSystem.cmake"
"CMakeLists.txt"
"pybind11/CMakeLists.txt"
"pybind11/tools/FindPythonLibsNew.cmake"
"pybind11/tools/pybind11Common.cmake"
"pybind11/tools/pybind11Tools.cmake"
"/usr/share/cmake-3.16/Modules/CMakeCInformation.cmake"
"/usr/share/cmake-3.16/Modules/CMakeCXXInformation.cmake"
"/usr/share/cmake-3.16/Modules/CMakeCheckCompilerFlagCommonPatterns.cmake"
"/usr/share/cmake-3.16/Modules/CMakeCommonLanguageInclude.cmake"
"/usr/share/cmake-3.16/Modules/CMakeDependentOption.cmake"
"/usr/share/cmake-3.16/Modules/CMakeGenericSystem.cmake"
"/usr/share/cmake-3.16/Modules/CMakeInitializeConfigs.cmake"
"/usr/share/cmake-3.16/Modules/CMakeLanguageInformation.cmake"
"/usr/share/cmake-3.16/Modules/CMakePackageConfigHelpers.cmake"
"/usr/share/cmake-3.16/Modules/CMakeParseArguments.cmake"
"/usr/share/cmake-3.16/Modules/CMakeSystemSpecificInformation.cmake"
"/usr/share/cmake-3.16/Modules/CMakeSystemSpecificInitialize.cmake"
"/usr/share/cmake-3.16/Modules/CheckCXXCompilerFlag.cmake"
"/usr/share/cmake-3.16/Modules/CheckCXXSourceCompiles.cmake"
"/usr/share/cmake-3.16/Modules/Compiler/CMakeCommonCompilerMacros.cmake"
"/usr/share/cmake-3.16/Modules/Compiler/GNU-C.cmake"
"/usr/share/cmake-3.16/Modules/Compiler/GNU-CXX.cmake"
"/usr/share/cmake-3.16/Modules/Compiler/GNU.cmake"
"/usr/share/cmake-3.16/Modules/FindPackageHandleStandardArgs.cmake"
"/usr/share/cmake-3.16/Modules/FindPackageMessage.cmake"
"/usr/share/cmake-3.16/Modules/FindPythonInterp.cmake"
"/usr/share/cmake-3.16/Modules/GNUInstallDirs.cmake"
"/usr/share/cmake-3.16/Modules/Internal/CMakeCheckCompilerFlag.cmake"
"/usr/share/cmake-3.16/Modules/Platform/Linux-GNU-C.cmake"
"/usr/share/cmake-3.16/Modules/Platform/Linux-GNU-CXX.cmake"
"/usr/share/cmake-3.16/Modules/Platform/Linux-GNU.cmake"
"/usr/share/cmake-3.16/Modules/Platform/Linux.cmake"
"/usr/share/cmake-3.16/Modules/Platform/UnixPaths.cmake"
"/usr/share/cmake-3.16/Modules/WriteBasicConfigVersionFile.cmake"
)
# The corresponding makefile is:
set(CMAKE_MAKEFILE_OUTPUTS
"Makefile"
"CMakeFiles/cmake.check_cache"
)
# Byproducts of CMake generate step:
set(CMAKE_MAKEFILE_PRODUCTS
"CMakeFiles/CMakeDirectoryInformation.cmake"
"pybind11/CMakeFiles/CMakeDirectoryInformation.cmake"
)
# Dependency information for all targets:
set(CMAKE_DEPEND_INFO_FILES
"CMakeFiles/sample_cores.dir/DependInfo.cmake"
)
# CMAKE generated file: DO NOT EDIT!
# Generated by "Unix Makefiles" Generator, CMake Version 3.16
# Default target executed when no arguments are given to make.
default_target: all
.PHONY : default_target
#=============================================================================
# Special targets provided by cmake.
# Disable implicit rules so canonical targets will work.
.SUFFIXES:
# Remove some rules from gmake that .SUFFIXES does not remove.
SUFFIXES =
.SUFFIXES: .hpux_make_needs_suffix_list
# Suppress display of executed commands.
$(VERBOSE).SILENT:
# A target that is always out of date.
cmake_force:
.PHONY : cmake_force
#=============================================================================
# Set environment variables for the build.
# The shell in which to execute make rules.
SHELL = /bin/sh
# The CMake executable.
CMAKE_COMMAND = /usr/bin/cmake
# The command to remove a file.
RM = /usr/bin/cmake -E remove -f
# Escaping for special characters.
EQUALS = =
# The top-level source directory on which CMake was run.
CMAKE_SOURCE_DIR = /home/sxx/zlj/rpc_ps/Sample
# The top-level build directory on which CMake was run.
CMAKE_BINARY_DIR = /home/sxx/zlj/rpc_ps/Sample
#=============================================================================
# Directory level rules for the build root directory
# The main recursive "all" target.
all: CMakeFiles/sample_cores.dir/all
all: pybind11/all
.PHONY : all
# The main recursive "preinstall" target.
preinstall: pybind11/preinstall
.PHONY : preinstall
# The main recursive "clean" target.
clean: CMakeFiles/sample_cores.dir/clean
clean: pybind11/clean
.PHONY : clean
#=============================================================================
# Directory level rules for directory pybind11
# Recursive "all" directory target.
pybind11/all:
.PHONY : pybind11/all
# Recursive "preinstall" directory target.
pybind11/preinstall:
.PHONY : pybind11/preinstall
# Recursive "clean" directory target.
pybind11/clean:
.PHONY : pybind11/clean
#=============================================================================
# Target rules for target CMakeFiles/sample_cores.dir
# All Build rule for target.
CMakeFiles/sample_cores.dir/all:
$(MAKE) -f CMakeFiles/sample_cores.dir/build.make CMakeFiles/sample_cores.dir/depend
$(MAKE) -f CMakeFiles/sample_cores.dir/build.make CMakeFiles/sample_cores.dir/build
@$(CMAKE_COMMAND) -E cmake_echo_color --switch=$(COLOR) --progress-dir=/home/sxx/zlj/rpc_ps/Sample/CMakeFiles --progress-num=1,2 "Built target sample_cores"
.PHONY : CMakeFiles/sample_cores.dir/all
# Build rule for subdir invocation for target.
CMakeFiles/sample_cores.dir/rule: cmake_check_build_system
$(CMAKE_COMMAND) -E cmake_progress_start /home/sxx/zlj/rpc_ps/Sample/CMakeFiles 2
$(MAKE) -f CMakeFiles/Makefile2 CMakeFiles/sample_cores.dir/all
$(CMAKE_COMMAND) -E cmake_progress_start /home/sxx/zlj/rpc_ps/Sample/CMakeFiles 0
.PHONY : CMakeFiles/sample_cores.dir/rule
# Convenience name for target.
sample_cores: CMakeFiles/sample_cores.dir/rule
.PHONY : sample_cores
# clean rule for target.
CMakeFiles/sample_cores.dir/clean:
$(MAKE) -f CMakeFiles/sample_cores.dir/build.make CMakeFiles/sample_cores.dir/clean
.PHONY : CMakeFiles/sample_cores.dir/clean
#=============================================================================
# Special targets to cleanup operation of make.
# Special rule to run CMake to check the build system integrity.
# No rule that depends on this can have commands that come from listfiles
# because they might be regenerated.
cmake_check_build_system:
$(CMAKE_COMMAND) -S$(CMAKE_SOURCE_DIR) -B$(CMAKE_BINARY_DIR) --check-build-system CMakeFiles/Makefile.cmake 0
.PHONY : cmake_check_build_system
/home/sxx/zlj/rpc_ps/Sample/CMakeFiles/rebuild_cache.dir
/home/sxx/zlj/rpc_ps/Sample/CMakeFiles/edit_cache.dir
/home/sxx/zlj/rpc_ps/Sample/CMakeFiles/sample_cores.dir
/home/sxx/zlj/rpc_ps/Sample/pybind11/CMakeFiles/rebuild_cache.dir
/home/sxx/zlj/rpc_ps/Sample/pybind11/CMakeFiles/edit_cache.dir
# This file is generated by cmake for dependency checking of the CMakeCache.txt file
# The set of languages for which implicit dependencies are needed:
set(CMAKE_DEPENDS_LANGUAGES
"CXX"
)
# The set of files for implicit dependencies of each language:
set(CMAKE_DEPENDS_CHECK_CXX
"/home/sxx/zlj/rpc_ps/Sample/sample_cores.cpp" "/home/sxx/zlj/rpc_ps/Sample/CMakeFiles/sample_cores.dir/sample_cores.cpp.o"
)
set(CMAKE_CXX_COMPILER_ID "GNU")
# Preprocessor definitions for this target.
set(CMAKE_TARGET_DEFINITIONS_CXX
"sample_cores_EXPORTS"
)
# The include file search paths:
set(CMAKE_CXX_TARGET_INCLUDE_PATH
"pybind11/include"
"/home/sxx/miniconda3/envs/gnn/include/python3.8"
)
# Targets to which this target links.
set(CMAKE_TARGET_LINKED_INFO_FILES
)
# Fortran module output directory.
set(CMAKE_Fortran_TARGET_MODULE_DIR "")
# CMAKE generated file: DO NOT EDIT!
# Generated by "Unix Makefiles" Generator, CMake Version 3.16
# Delete rule output on recipe failure.
.DELETE_ON_ERROR:
#=============================================================================
# Special targets provided by cmake.
# Disable implicit rules so canonical targets will work.
.SUFFIXES:
# Remove some rules from gmake that .SUFFIXES does not remove.
SUFFIXES =
.SUFFIXES: .hpux_make_needs_suffix_list
# Suppress display of executed commands.
$(VERBOSE).SILENT:
# A target that is always out of date.
cmake_force:
.PHONY : cmake_force
#=============================================================================
# Set environment variables for the build.
# The shell in which to execute make rules.
SHELL = /bin/sh
# The CMake executable.
CMAKE_COMMAND = /usr/bin/cmake
# The command to remove a file.
RM = /usr/bin/cmake -E remove -f
# Escaping for special characters.
EQUALS = =
# The top-level source directory on which CMake was run.
CMAKE_SOURCE_DIR = /home/sxx/zlj/rpc_ps/Sample
# The top-level build directory on which CMake was run.
CMAKE_BINARY_DIR = /home/sxx/zlj/rpc_ps/Sample
# Include any dependencies generated for this target.
include CMakeFiles/sample_cores.dir/depend.make
# Include the progress variables for this target.
include CMakeFiles/sample_cores.dir/progress.make
# Include the compile flags for this target's objects.
include CMakeFiles/sample_cores.dir/flags.make
CMakeFiles/sample_cores.dir/sample_cores.cpp.o: CMakeFiles/sample_cores.dir/flags.make
CMakeFiles/sample_cores.dir/sample_cores.cpp.o: sample_cores.cpp
@$(CMAKE_COMMAND) -E cmake_echo_color --switch=$(COLOR) --green --progress-dir=/home/sxx/zlj/rpc_ps/Sample/CMakeFiles --progress-num=$(CMAKE_PROGRESS_1) "Building CXX object CMakeFiles/sample_cores.dir/sample_cores.cpp.o"
/usr/bin/c++ $(CXX_DEFINES) $(CXX_INCLUDES) $(CXX_FLAGS) -o CMakeFiles/sample_cores.dir/sample_cores.cpp.o -c /home/sxx/zlj/rpc_ps/Sample/sample_cores.cpp
CMakeFiles/sample_cores.dir/sample_cores.cpp.i: cmake_force
@$(CMAKE_COMMAND) -E cmake_echo_color --switch=$(COLOR) --green "Preprocessing CXX source to CMakeFiles/sample_cores.dir/sample_cores.cpp.i"
/usr/bin/c++ $(CXX_DEFINES) $(CXX_INCLUDES) $(CXX_FLAGS) -E /home/sxx/zlj/rpc_ps/Sample/sample_cores.cpp > CMakeFiles/sample_cores.dir/sample_cores.cpp.i
CMakeFiles/sample_cores.dir/sample_cores.cpp.s: cmake_force
@$(CMAKE_COMMAND) -E cmake_echo_color --switch=$(COLOR) --green "Compiling CXX source to assembly CMakeFiles/sample_cores.dir/sample_cores.cpp.s"
/usr/bin/c++ $(CXX_DEFINES) $(CXX_INCLUDES) $(CXX_FLAGS) -S /home/sxx/zlj/rpc_ps/Sample/sample_cores.cpp -o CMakeFiles/sample_cores.dir/sample_cores.cpp.s
# Object files for target sample_cores
sample_cores_OBJECTS = \
"CMakeFiles/sample_cores.dir/sample_cores.cpp.o"
# External object files for target sample_cores
sample_cores_EXTERNAL_OBJECTS =
sample_cores.cpython-38-x86_64-linux-gnu.so: CMakeFiles/sample_cores.dir/sample_cores.cpp.o
sample_cores.cpython-38-x86_64-linux-gnu.so: CMakeFiles/sample_cores.dir/build.make
sample_cores.cpython-38-x86_64-linux-gnu.so: CMakeFiles/sample_cores.dir/link.txt
@$(CMAKE_COMMAND) -E cmake_echo_color --switch=$(COLOR) --green --bold --progress-dir=/home/sxx/zlj/rpc_ps/Sample/CMakeFiles --progress-num=$(CMAKE_PROGRESS_2) "Linking CXX shared module sample_cores.cpython-38-x86_64-linux-gnu.so"
$(CMAKE_COMMAND) -E cmake_link_script CMakeFiles/sample_cores.dir/link.txt --verbose=$(VERBOSE)
# Rule to build all files generated by this target.
CMakeFiles/sample_cores.dir/build: sample_cores.cpython-38-x86_64-linux-gnu.so
.PHONY : CMakeFiles/sample_cores.dir/build
CMakeFiles/sample_cores.dir/clean:
$(CMAKE_COMMAND) -P CMakeFiles/sample_cores.dir/cmake_clean.cmake
.PHONY : CMakeFiles/sample_cores.dir/clean
CMakeFiles/sample_cores.dir/depend:
cd /home/sxx/zlj/rpc_ps/Sample && $(CMAKE_COMMAND) -E cmake_depends "Unix Makefiles" /home/sxx/zlj/rpc_ps/Sample /home/sxx/zlj/rpc_ps/Sample /home/sxx/zlj/rpc_ps/Sample /home/sxx/zlj/rpc_ps/Sample /home/sxx/zlj/rpc_ps/Sample/CMakeFiles/sample_cores.dir/DependInfo.cmake --color=$(COLOR)
.PHONY : CMakeFiles/sample_cores.dir/depend
file(REMOVE_RECURSE
"CMakeFiles/sample_cores.dir/sample_cores.cpp.o"
"sample_cores.cpython-38-x86_64-linux-gnu.so"
"sample_cores.pdb"
)
# Per-language clean rules from dependency scanning.
foreach(lang CXX)
include(CMakeFiles/sample_cores.dir/cmake_clean_${lang}.cmake OPTIONAL)
endforeach()
# CMAKE generated file: DO NOT EDIT!
# Generated by "Unix Makefiles" Generator, CMake Version 3.16
CMakeFiles/sample_cores.dir/sample_cores.cpp.o
/home/sxx/miniconda3/envs/gnn/include/python3.8/Python.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/abstract.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/bltinmodule.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/boolobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/bytearrayobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/bytesobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/cellobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/ceval.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/classobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/code.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/codecs.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/compile.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/complexobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/context.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/cpython/abstract.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/cpython/dictobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/cpython/fileobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/cpython/initconfig.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/cpython/object.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/cpython/objimpl.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/cpython/pyerrors.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/cpython/pylifecycle.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/cpython/pymem.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/cpython/pystate.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/cpython/sysmodule.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/cpython/traceback.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/cpython/tupleobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/cpython/unicodeobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/descrobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/dictobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/dtoa.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/enumobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/eval.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/fileobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/fileutils.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/floatobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/frameobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/funcobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/genobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/import.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/intrcheck.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/iterobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/listobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/longintrepr.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/longobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/memoryobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/methodobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/modsupport.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/moduleobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/namespaceobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/object.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/objimpl.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/odictobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/osmodule.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/patchlevel.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/picklebufobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/pyarena.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/pycapsule.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/pyconfig.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/pyctype.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/pydebug.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/pyerrors.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/pyfpe.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/pyhash.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/pylifecycle.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/pymacconfig.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/pymacro.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/pymath.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/pymem.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/pyport.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/pystate.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/pystrcmp.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/pystrtod.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/pythonrun.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/pythread.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/pytime.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/rangeobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/setobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/sliceobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/structseq.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/sysmodule.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/traceback.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/tracemalloc.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/tupleobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/typeslots.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/unicodeobject.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/warnings.h
/home/sxx/miniconda3/envs/gnn/include/python3.8/weakrefobject.h
/home/sxx/zlj/rpc_ps/Sample/sample_cores.cpp
pybind11/include/pybind11/attr.h
pybind11/include/pybind11/buffer_info.h
pybind11/include/pybind11/cast.h
pybind11/include/pybind11/complex.h
pybind11/include/pybind11/detail/class.h
pybind11/include/pybind11/detail/common.h
pybind11/include/pybind11/detail/descr.h
pybind11/include/pybind11/detail/init.h
pybind11/include/pybind11/detail/internals.h
pybind11/include/pybind11/detail/type_caster_base.h
pybind11/include/pybind11/detail/typeid.h
pybind11/include/pybind11/gil.h
pybind11/include/pybind11/numpy.h
pybind11/include/pybind11/options.h
pybind11/include/pybind11/pybind11.h
pybind11/include/pybind11/pytypes.h
pybind11/include/pybind11/stl.h
# CMAKE generated file: DO NOT EDIT!
# Generated by "Unix Makefiles" Generator, CMake Version 3.16
# compile CXX with /usr/bin/c++
CXX_FLAGS = -fPIC -fvisibility=hidden -flto -fno-fat-lto-objects
CXX_DEFINES = -Dsample_cores_EXPORTS
CXX_INCLUDES = -isystem /home/sxx/zlj/rpc_ps/Sample/pybind11/include -isystem /home/sxx/miniconda3/envs/gnn/include/python3.8
/usr/bin/c++ -fPIC -flto -shared -o sample_cores.cpython-38-x86_64-linux-gnu.so CMakeFiles/sample_cores.dir/sample_cores.cpp.o
CMAKE_PROGRESS_1 = 1
CMAKE_PROGRESS_2 = 2
cmake_minimum_required(VERSION 2.8.12)
project(sample_cores)
add_subdirectory(pybind11)
pybind11_add_module(sample_cores sample_cores.cpp)
# CMAKE generated file: DO NOT EDIT!
# Generated by "Unix Makefiles" Generator, CMake Version 3.16
# Default target executed when no arguments are given to make.
default_target: all
.PHONY : default_target
# Allow only one "make -f Makefile2" at a time, but pass parallelism.
.NOTPARALLEL:
#=============================================================================
# Special targets provided by cmake.
# Disable implicit rules so canonical targets will work.
.SUFFIXES:
# Remove some rules from gmake that .SUFFIXES does not remove.
SUFFIXES =
.SUFFIXES: .hpux_make_needs_suffix_list
# Suppress display of executed commands.
$(VERBOSE).SILENT:
# A target that is always out of date.
cmake_force:
.PHONY : cmake_force
#=============================================================================
# Set environment variables for the build.
# The shell in which to execute make rules.
SHELL = /bin/sh
# The CMake executable.
CMAKE_COMMAND = /usr/bin/cmake
# The command to remove a file.
RM = /usr/bin/cmake -E remove -f
# Escaping for special characters.
EQUALS = =
# The top-level source directory on which CMake was run.
CMAKE_SOURCE_DIR = /home/sxx/zlj/rpc_ps/Sample
# The top-level build directory on which CMake was run.
CMAKE_BINARY_DIR = /home/sxx/zlj/rpc_ps/Sample
#=============================================================================
# Targets provided globally by CMake.
# Special rule for the target rebuild_cache
rebuild_cache:
@$(CMAKE_COMMAND) -E cmake_echo_color --switch=$(COLOR) --cyan "Running CMake to regenerate build system..."
/usr/bin/cmake -S$(CMAKE_SOURCE_DIR) -B$(CMAKE_BINARY_DIR)
.PHONY : rebuild_cache
# Special rule for the target rebuild_cache
rebuild_cache/fast: rebuild_cache
.PHONY : rebuild_cache/fast
# Special rule for the target edit_cache
edit_cache:
@$(CMAKE_COMMAND) -E cmake_echo_color --switch=$(COLOR) --cyan "No interactive CMake dialog available..."
/usr/bin/cmake -E echo No\ interactive\ CMake\ dialog\ available.
.PHONY : edit_cache
# Special rule for the target edit_cache
edit_cache/fast: edit_cache
.PHONY : edit_cache/fast
# The main all target
all: cmake_check_build_system
$(CMAKE_COMMAND) -E cmake_progress_start /home/sxx/zlj/rpc_ps/Sample/CMakeFiles /home/sxx/zlj/rpc_ps/Sample/CMakeFiles/progress.marks
$(MAKE) -f CMakeFiles/Makefile2 all
$(CMAKE_COMMAND) -E cmake_progress_start /home/sxx/zlj/rpc_ps/Sample/CMakeFiles 0
.PHONY : all
# The main clean target
clean:
$(MAKE) -f CMakeFiles/Makefile2 clean
.PHONY : clean
# The main clean target
clean/fast: clean
.PHONY : clean/fast
# Prepare targets for installation.
preinstall: all
$(MAKE) -f CMakeFiles/Makefile2 preinstall
.PHONY : preinstall
# Prepare targets for installation.
preinstall/fast:
$(MAKE) -f CMakeFiles/Makefile2 preinstall
.PHONY : preinstall/fast
# clear depends
depend:
$(CMAKE_COMMAND) -S$(CMAKE_SOURCE_DIR) -B$(CMAKE_BINARY_DIR) --check-build-system CMakeFiles/Makefile.cmake 1
.PHONY : depend
#=============================================================================
# Target rules for targets named sample_cores
# Build rule for target.
sample_cores: cmake_check_build_system
$(MAKE) -f CMakeFiles/Makefile2 sample_cores
.PHONY : sample_cores
# fast build rule for target.
sample_cores/fast:
$(MAKE) -f CMakeFiles/sample_cores.dir/build.make CMakeFiles/sample_cores.dir/build
.PHONY : sample_cores/fast
sample_cores.o: sample_cores.cpp.o
.PHONY : sample_cores.o
# target to build an object file
sample_cores.cpp.o:
$(MAKE) -f CMakeFiles/sample_cores.dir/build.make CMakeFiles/sample_cores.dir/sample_cores.cpp.o
.PHONY : sample_cores.cpp.o
sample_cores.i: sample_cores.cpp.i
.PHONY : sample_cores.i
# target to preprocess a source file
sample_cores.cpp.i:
$(MAKE) -f CMakeFiles/sample_cores.dir/build.make CMakeFiles/sample_cores.dir/sample_cores.cpp.i
.PHONY : sample_cores.cpp.i
sample_cores.s: sample_cores.cpp.s
.PHONY : sample_cores.s
# target to generate assembly for a file
sample_cores.cpp.s:
$(MAKE) -f CMakeFiles/sample_cores.dir/build.make CMakeFiles/sample_cores.dir/sample_cores.cpp.s
.PHONY : sample_cores.cpp.s
# Help Target
help:
@echo "The following are some of the valid targets for this Makefile:"
@echo "... all (the default if no target is provided)"
@echo "... clean"
@echo "... depend"
@echo "... rebuild_cache"
@echo "... edit_cache"
@echo "... sample_cores"
@echo "... sample_cores.o"
@echo "... sample_cores.i"
@echo "... sample_cores.s"
.PHONY : help
#=============================================================================
# Special targets to cleanup operation of make.
# Special rule to run CMake to check the build system integrity.
# No rule that depends on this can have commands that come from listfiles
# because they might be regenerated.
cmake_check_build_system:
$(CMAKE_COMMAND) -S$(CMAKE_SOURCE_DIR) -B$(CMAKE_BINARY_DIR) --check-build-system CMakeFiles/Makefile.cmake 0
.PHONY : cmake_check_build_system
# Install script for directory: /home/sxx/zlj/rpc_ps/Sample
# Set the install prefix
if(NOT DEFINED CMAKE_INSTALL_PREFIX)
set(CMAKE_INSTALL_PREFIX "/usr/local")
endif()
string(REGEX REPLACE "/$" "" CMAKE_INSTALL_PREFIX "${CMAKE_INSTALL_PREFIX}")
# Set the install configuration name.
if(NOT DEFINED CMAKE_INSTALL_CONFIG_NAME)
if(BUILD_TYPE)
string(REGEX REPLACE "^[^A-Za-z0-9_]+" ""
CMAKE_INSTALL_CONFIG_NAME "${BUILD_TYPE}")
else()
set(CMAKE_INSTALL_CONFIG_NAME "")
endif()
message(STATUS "Install configuration: \"${CMAKE_INSTALL_CONFIG_NAME}\"")
endif()
# Set the component getting installed.
if(NOT CMAKE_INSTALL_COMPONENT)
if(COMPONENT)
message(STATUS "Install component: \"${COMPONENT}\"")
set(CMAKE_INSTALL_COMPONENT "${COMPONENT}")
else()
set(CMAKE_INSTALL_COMPONENT)
endif()
endif()
# Install shared libraries without execute permission?
if(NOT DEFINED CMAKE_INSTALL_SO_NO_EXE)
set(CMAKE_INSTALL_SO_NO_EXE "1")
endif()
# Is this installation the result of a crosscompile?
if(NOT DEFINED CMAKE_CROSSCOMPILING)
set(CMAKE_CROSSCOMPILING "FALSE")
endif()
if(NOT CMAKE_INSTALL_LOCAL_ONLY)
# Include the install script for each subdirectory.
include("/home/sxx/zlj/rpc_ps/Sample/pybind11/cmake_install.cmake")
endif()
if(CMAKE_INSTALL_COMPONENT)
set(CMAKE_INSTALL_MANIFEST "install_manifest_${CMAKE_INSTALL_COMPONENT}.txt")
else()
set(CMAKE_INSTALL_MANIFEST "install_manifest.txt")
endif()
string(REPLACE ";" "\n" CMAKE_INSTALL_MANIFEST_CONTENT
"${CMAKE_INSTALL_MANIFEST_FILES}")
file(WRITE "/home/sxx/zlj/rpc_ps/Sample/${CMAKE_INSTALL_MANIFEST}"
"${CMAKE_INSTALL_MANIFEST_CONTENT}")
import torch import torch
# import sys
# from os.path import abspath, dirname
# sys.path.insert(0, abspath(dirname(__file__)))
# print(sys.path)
from neighbor_sampler import NeighborSampler from neighbor_sampler import NeighborSampler
# edge_index = torch.tensor([[0, 1, 1, 2, 2, 2, 3], [1, 0, 2, 1, 3, 0, 2]]) # edge_index = torch.tensor([[0, 1, 1, 2, 2, 2, 3], [1, 0, 2, 1, 3, 0, 2]])
......
...@@ -5,8 +5,7 @@ import math ...@@ -5,8 +5,7 @@ import math
import torch import torch
import torch.multiprocessing as mp import torch.multiprocessing as mp
from typing import Optional, Tuple from typing import Optional, Tuple
#import ctypes
#sample_lib = ctypes.cdll.LoadLibrary('Sample/sample_cores.cpython-37m-x86_64-linux-gnu.so')
from base import BaseSampler, NegativeSampling from base import BaseSampler, NegativeSampling
from sample_cores import get_neighbors, neighbor_sample_from_nodes, heads_unique from sample_cores import get_neighbors, neighbor_sample_from_nodes, heads_unique
...@@ -17,17 +16,20 @@ class NeighborSampler(BaseSampler): ...@@ -17,17 +16,20 @@ class NeighborSampler(BaseSampler):
num_layers: int, num_layers: int,
fanout: list, fanout: list,
workers = 1, workers = 1,
edge_index : torch.Tensor = None, edge_index : Optional[torch.Tensor] = None,
deg = None, deg = None,
neighbors = None neighbors = None
) -> None: ) -> None:
r"""__init__ r"""__init__
Args: Args:
edge_index: all edges in the graph
num_nodes: the num of all nodes in the graph num_nodes: the num of all nodes in the graph
num_layers: the num of layers to be sampled num_layers: the num of layers to be sampled
fanout: the list of max neighbors' number chosen for each layer fanout: the list of max neighbors' number chosen for each layer
workers: the number of threads, default value is 1 workers: the number of threads, default value is 1
edge_index: all edges in the graph
neighbors: all nodes' neighbors
deg: the degree of all nodes
should provide edge_index or (neighbors, deg)
""" """
super().__init__() super().__init__()
self.num_layers = num_layers self.num_layers = num_layers
...@@ -123,6 +125,7 @@ class NeighborSampler(BaseSampler): ...@@ -123,6 +125,7 @@ class NeighborSampler(BaseSampler):
if neg_sampling is None or neg_sampling.is_binary(): if neg_sampling is None or neg_sampling.is_binary():
edge_label_index = inverse_seed.view(2, -1) edge_label_index = inverse_seed.view(2, -1)
# edge_label_index不知道是啥 edge_label是seed links的标签
metadata = {'edge_label_index':edge_label_index, 'edge_label':edge_label} metadata = {'edge_label_index':edge_label_index, 'edge_label':edge_label}
elif neg_sampling.is_triplet(): elif neg_sampling.is_triplet():
...@@ -130,8 +133,11 @@ class NeighborSampler(BaseSampler): ...@@ -130,8 +133,11 @@ class NeighborSampler(BaseSampler):
dst_pos_index = inverse_seed[num_pos:2 * num_pos] dst_pos_index = inverse_seed[num_pos:2 * num_pos]
dst_neg_index = inverse_seed[2 * num_pos:] dst_neg_index = inverse_seed[2 * num_pos:]
dst_neg_index = dst_neg_index.view(num_pos, -1).squeeze(-1) dst_neg_index = dst_neg_index.view(num_pos, -1).squeeze(-1)
# src_index是seed里src点的索引
# dst_pos_index是seed里dst_pos点的索引
# dst_neg_index是seed里dst_neg点的索引
metadata = {'src_index':src_index, 'dst_pos_index':dst_pos_index, 'dst_neg_index':dst_neg_index} metadata = {'src_index':src_index, 'dst_pos_index':dst_pos_index, 'dst_neg_index':dst_neg_index}
# sampled_nodes最前方是原始序列的采样起点也就是去重后的seed
return sampled_nodes, sampled_edge_index_list, metadata return sampled_nodes, sampled_edge_index_list, metadata
def _sample_one_layer_from_nodes( def _sample_one_layer_from_nodes(
......
...@@ -172,6 +172,6 @@ PYBIND11_MODULE(sample_cores, m) ...@@ -172,6 +172,6 @@ PYBIND11_MODULE(sample_cores, m)
py::class_<TemporalNeighborBlock>(m, "TemporalNeighborBlock") py::class_<TemporalNeighborBlock>(m, "TemporalNeighborBlock")
.def(py::init<std::vector<vector<NodeIDType>*>&, .def(py::init<std::vector<vector<NodeIDType>*>&,
std::vector<int> &>()) std::vector<int> &>())
.def_readonly("neighbors", &TemporalNeighborBlock::neighbors) .def_readonly("neighbors", &TemporalNeighborBlock::neighbors, py::return_value_policy::copy)
.def_readonly("deg", &TemporalNeighborBlock::deg); .def_readonly("deg", &TemporalNeighborBlock::deg, py::return_value_policy::copy);
} }
pybind11 @ 0e01c243
Subproject commit 0e01c243c7ffae3a2e52f998bacfe82f56aa96d9
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment