mir/criu
2
0
Fork 0
mirror of https://github.com/checkpoint-restore/criu synced 2025-09-02 23:35:21 +00:00
Code Issues Releases Wiki Activity

criu/plugin: Add unit tests for GPU remapping

Browse Source 
Adding unit tests for GPU remapping code when checkpointing and
restoring on different nodes with different topologies.

Signed-off-by: David Yat Sin <david.yatsin@amd.com>
This commit is contained in:
David Yat Sin
2021-05-18 12:15:48 -04:00
committed by Andrei Vagin
parent 4856e0d4d0
commit ba9c62df24
3 changed files with 1152 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
plugins/amdgpu/Makefile
View File

@@ -5,7 +5,7 @@ PLUGIN_INC := ../../../criu/include
PLUGIN_INC_EXTRA := ../../criu/include PLUGIN_INC_EXTRA := ../../criu/include
PLUGIN_INCLUDE := -iquote$(PLUGIN_INC) -iquote$(PLUGIN_INC_EXTRA) PLUGIN_INCLUDE := -iquote$(PLUGIN_INC) -iquote$(PLUGIN_INC_EXTRA)
LIBDRM_INC := -I/usr/include/libdrm LIBDRM_INC := -I/usr/include/libdrm
DEPS_OK := amdgpu_plugin.so DEPS_OK := amdgpu_plugin.so amdgpu_plugin_test
DEPS_NOK := ; DEPS_NOK := ;
include $(__nmk_dir)msg.mk include $(__nmk_dir)msg.mk
@@ -29,7 +29,18 @@ amdgpu_plugin_clean:
$(call msg-clean, $@) $(call msg-clean, $@)
$(Q) $(RM) amdgpu_plugin.so criu-amdgpu.pb-c* $(Q) $(RM) amdgpu_plugin.so criu-amdgpu.pb-c*
.PHONY: amdgpu_plugin_clean .PHONY: amdgpu_plugin_clean
clean: amdgpu_plugin_clean
test_topology_remap: amdgpu_plugin_topology.c tests/test_topology_remap.c
$(CC) $^ -o $@ -DCOMPILE_TESTS $(PLUGIN_INCLUDE) -I .
amdgpu_plugin_test: test_topology_remap
.PHONY: amdgpu_plugin_test
amdgpu_plugin_test_clean:
$(Q) $(RM) test_topology_remap
.PHONY: amdgpu_plugin_test_clean
clean: amdgpu_plugin_clean amdgpu_plugin_test_clean
mrproper: clean mrproper: clean

27
plugins/amdgpu/amdgpu_plugin_topology.c
View File

@@ -24,6 +24,19 @@
#define _GNU_SOURCE 1 #define _GNU_SOURCE 1
#endif #endif
#ifdef COMPILE_TESTS
#undef pr_err
#define pr_err(format, arg...) fprintf(stdout, "%s:%d ERROR:" format, __FILE__, __LINE__, ##arg)
#undef pr_info
#define pr_info(format, arg...) fprintf(stdout, "%s:%d INFO:" format, __FILE__, __LINE__, ##arg)
#undef pr_debug
#define pr_debug(format, arg...) fprintf(stdout, "%s:%d DBG:" format, __FILE__, __LINE__, ##arg)
#undef pr_perror
#define pr_perror(format, arg...) \
fprintf(stdout, "%s:%d: " format " (errno = %d (%s))\n", __FILE__, __LINE__, ##arg, errno, strerror(errno))
#endif
#ifdef DEBUG #ifdef DEBUG
#define plugin_log_msg(fmt, ...) pr_debug(fmt, ##__VA_ARGS__) #define plugin_log_msg(fmt, ...) pr_debug(fmt, ##__VA_ARGS__)
#else #else
@@ -34,19 +47,19 @@
/* User override options */ /* User override options */
/* Skip firmware version check */ /* Skip firmware version check */
bool kfd_fw_version_check; bool kfd_fw_version_check = true;
/* Skip SDMA firmware version check */ /* Skip SDMA firmware version check */
bool kfd_sdma_fw_version_check; bool kfd_sdma_fw_version_check = true;
/* Skip caches count check */ /* Skip caches count check */
bool kfd_caches_count_check; bool kfd_caches_count_check = true;
/* Skip num gws check */ /* Skip num gws check */
bool kfd_num_gws_check; bool kfd_num_gws_check = true;
/* Skip vram size check */ /* Skip vram size check */
bool kfd_vram_size_check; bool kfd_vram_size_check = true;
/* Preserve NUMA regions */ /* Preserve NUMA regions */
bool kfd_numa_check; bool kfd_numa_check = true;
/* Skip capability check */ /* Skip capability check */
bool kfd_capability_check; bool kfd_capability_check = true;
static int open_drm_render_device(int minor) static int open_drm_render_device(int minor)
{ {

1119
plugins/amdgpu/tests/test_topology_remap.c Normal file
View File

File diff suppressed because it is too large Load Diff