AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add fork support

Browse Source 
If fork() is called, clear all duplicated data that is invalid in the
child process.

Change-Id: I4e27198060db593c630c6337b7071dfbd0d80b83
Signed-off-by: Harish Kasiviswanathan <Harish.Kasiviswanathan@amd.com>


[ROCm/ROCR-Runtime commit: f1f62d863c]
This commit is contained in:
Harish Kasiviswanathan
2017-01-10 14:34:47 -05:00
parent f07e9a5606
commit 6f2b43d96e
5 changed files with 120 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocr-runtime/src/openclose.c
+25 -6
View File
@@ -36,20 +36,39 @@
static const char kfd_device_name[] = "/dev/kfd";
static const char tmp_file[] = "/var/lock/.amd_hsa_thunk_lock";
int amd_hsa_thunk_lock_fd = 0;
static pid_t parent_pid = -1;
static bool is_forked_child(void)
{
static pid_t open_pid = -1;
pid_t my_pid = getpid();
pid_t cur_pid = getpid();
if (open_pid == -1 || open_pid != my_pid) {
open_pid = my_pid;
return true;
if (parent_pid == -1) {
parent_pid = cur_pid;
return false;
}
if (parent_pid != cur_pid)
return true;
return false;
}
/* Call this from the child process after fork. This will clear all
* data that is duplicated from the parent process, that is not valid
* in the child.
* The topology information is duplicated from the parent is valid
* in the child process so it is not cleared
*/
static void clear_after_fork(void)
{
clear_process_doorbells();
clear_events_page();
fmm_clear_all_mem();
destroy_device_debugging_memory();
kfd_open_count = 0;
}
HSAKMT_STATUS
HSAKMTAPI
hsaKmtOpenKFD(void)
@@ -65,7 +84,7 @@ hsaKmtOpenKFD(void)
* belong to the parent
*/
if (is_forked_child())
kfd_open_count = 0;
clear_after_fork();
if (kfd_open_count == 0)
{