From patchwork Fri Oct 4 10:13:50 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Georgii Burunsuzian X-Patchwork-Id: 1992675 Return-Path: X-Original-To: incoming@patchwork.ozlabs.org Delivered-To: patchwork-incoming@legolas.ozlabs.org Authentication-Results: legolas.ozlabs.org; dkim=fail reason="signature verification failed" (2048-bit key; unprotected) header.d=gmail.com header.i=@gmail.com header.a=rsa-sha256 header.s=20230601 header.b=S819Ir3D; dkim-atps=neutral Authentication-Results: legolas.ozlabs.org; spf=pass (sender SPF authorized) smtp.mailfrom=gcc.gnu.org (client-ip=8.43.85.97; helo=server2.sourceware.org; envelope-from=gcc-patches-bounces~incoming=patchwork.ozlabs.org@gcc.gnu.org; receiver=patchwork.ozlabs.org) Received: from server2.sourceware.org (server2.sourceware.org [8.43.85.97]) (using TLSv1.3 with cipher TLS_AES_256_GCM_SHA384 (256/256 bits) key-exchange X25519 server-signature ECDSA (secp384r1) server-digest SHA384) (No client certificate requested) by legolas.ozlabs.org (Postfix) with ESMTPS id 4XKktM2hP4z1xt7 for ; Fri, 4 Oct 2024 20:14:35 +1000 (AEST) Received: from server2.sourceware.org (localhost [IPv6:::1]) by sourceware.org (Postfix) with ESMTP id 8FF0A3844777 for ; Fri, 4 Oct 2024 10:14:33 +0000 (GMT) X-Original-To: gcc-patches@gcc.gnu.org Delivered-To: gcc-patches@gcc.gnu.org Received: from mail-yw1-x112c.google.com (mail-yw1-x112c.google.com [IPv6:2607:f8b0:4864:20::112c]) by sourceware.org (Postfix) with ESMTPS id C659D385DDD7 for ; Fri, 4 Oct 2024 10:14:01 +0000 (GMT) DMARC-Filter: OpenDMARC Filter v1.4.2 sourceware.org C659D385DDD7 Authentication-Results: sourceware.org; dmarc=pass (p=none dis=none) header.from=gmail.com Authentication-Results: sourceware.org; spf=pass smtp.mailfrom=gmail.com ARC-Filter: OpenARC Filter v1.0.0 sourceware.org C659D385DDD7 Authentication-Results: server2.sourceware.org; arc=none smtp.remote-ip=2607:f8b0:4864:20::112c ARC-Seal: i=1; a=rsa-sha256; d=sourceware.org; s=key; t=1728036846; cv=none; b=AQcLXPR9sv/wEYIV3EiWqmPF+7jFtl2fmnkMzN2Jcfv/TPSSyb/NSJFjzNHa+rTfWTrLvx+HOwCQH5cLTEinHLMQnw7aVDXCaqKA8l0GRWcmVSYkYDr2ToYCU3rqP5jG69K1snIy+BdtitRwEIYTbb/vBB2XNEnZYSm2ikQCZ1c= ARC-Message-Signature: i=1; a=rsa-sha256; d=sourceware.org; s=key; t=1728036846; c=relaxed/simple; bh=gYt3yYgcLlPucaCZ8c7E6DU9luVqhijx7EnuOZHAJ/E=; h=DKIM-Signature:MIME-Version:From:Date:Message-ID:Subject:To; b=MZNCuCYdSOxJfrqo2XTyF1iiMDsrV4InD6T+aXhCq5quOETRRbE+P21nlTXVHOPBvvuYv9Go887xhNsHfRCkyUzrREa1CtOHaKVUJ0VnUGeaDaNI3kauR5jS6AbA4R05RPhBeJSlCMt8qxkS79fz14bj72gL231ffOoLLO0xQKY= ARC-Authentication-Results: i=1; server2.sourceware.org Received: by mail-yw1-x112c.google.com with SMTP id 00721157ae682-6e25f3748e0so21029687b3.0 for ; Fri, 04 Oct 2024 03:14:01 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20230601; t=1728036841; x=1728641641; darn=gcc.gnu.org; h=to:subject:message-id:date:from:mime-version:from:to:cc:subject :date:message-id:reply-to; bh=k+PR65b6hdWxCsYdd0brm8HUiCKOg0p9K2mKhD4QrEk=; b=S819Ir3D5XzMUcMS53wrypXxnpRFyuUR6syLa1/T8ZPqHZXYKavL/itLgoBZJXl/FJ FKUMpeTFV0bSkXJcfrNInuHm7dDr1dy0Y8io4hRYfQ75ywmzF0+s3/4iqgvzVGEKot6q tv55fQMwtEaUtsIveieyRJs5jVL54y2Ysxd9WeE9sUlgEtuYsouggrBh4O54T62S9Ym0 /OTzkB8r1jFWJSzKVdw63ddp0U+sTxcWudS5FhS0s6v1NETWER0xn+bqk5II7r/7zKkz hZHMmXoLkddnxL0/bvEXHj14M97Ub5JAOS3VzJBXap3AcLKVBL0QQ087s8IfJmvINciO dfKQ== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1728036841; x=1728641641; h=to:subject:message-id:date:from:mime-version:x-gm-message-state :from:to:cc:subject:date:message-id:reply-to; bh=k+PR65b6hdWxCsYdd0brm8HUiCKOg0p9K2mKhD4QrEk=; b=v3uK5+/2FY/rL5rhvWwd6TBYLo50IvIOMT6qy9dTedmqFT8BCqK11GOre/CMWTR26C 84rqS8/tY1jUYtDJtAGy3jxFUgtQMiU5XGGzNfwSeuY+hDe8hdWKAC0DRXUH1Dh/FLy8 +H1IeFjPVckShQPSXzMM/y6vhvwHxeg1oq0UnRyqOA5JevkYGxLKfOEU9EttCP+JVmbn mowd4WhNMHB54ZsPJYuPMHhFPm0uwvbz1/9HdWZ3A3OKSQD+Fw9I0Dbwx2LUyth40grM CBe3i4rW6xI5zX95ZCLGNEW/OFSoMRf8eDBxJr8SV8OPY1OOEti88Zqr46g6/o/McvlE MI+A== X-Gm-Message-State: AOJu0YzQTjiPJTr7VDuAEV/KB7Zyt/+hQyd7iQXxWbeoNhVueI3dzPBf 84rcfjMtgP/Zi+J17y81ACAviD/QPq6O4ZkGMm8rG3IIyNeJZAizyqBBEFxVONYLfN2R9PxdzGs fAYmrMPNPgsWmmkBn2c637gYJALik8K9qFhM= X-Google-Smtp-Source: AGHT+IEsOKrBIhsbB6E4nrhSNh0id3Q7ULfzIHQN9xmB2WY1JMrIm9QTujem6J52xro/5GWCUTSr1aB8vPSI4KNpdD0= X-Received: by 2002:a05:690c:281:b0:6dd:bace:572 with SMTP id 00721157ae682-6e2c70297b7mr21871897b3.19.1728036840953; Fri, 04 Oct 2024 03:14:00 -0700 (PDT) MIME-Version: 1.0 From: Georgii Burunsuzian Date: Fri, 4 Oct 2024 13:13:50 +0300 Message-ID: Subject: [PATCH] WIP: Initial submission of libgomp host_process plugin To: gcc-patches@gcc.gnu.org X-Spam-Status: No, score=-8.4 required=5.0 tests=BAYES_00, DKIM_SIGNED, DKIM_VALID, DKIM_VALID_AU, DKIM_VALID_EF, FREEMAIL_ENVFROM_END_DIGIT, FREEMAIL_FROM, GIT_PATCH_0, HTML_MESSAGE, KAM_LINEPADDING, RCVD_IN_DNSWL_NONE, SPF_HELO_NONE, SPF_PASS, TXREP autolearn=ham autolearn_force=no version=3.4.6 X-Spam-Checker-Version: SpamAssassin 3.4.6 (2021-04-09) on server2.sourceware.org X-BeenThere: gcc-patches@gcc.gnu.org X-Mailman-Version: 2.1.30 Precedence: list List-Id: Gcc-patches mailing list List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Errors-To: gcc-patches-bounces~incoming=patchwork.ozlabs.org@gcc.gnu.org The first patch within the GSoC project "Offloading to a separate process on the same host". The project focuses on streamlining the debugging and development process for code intended for offloading to devices like GPUs. It proposes a method to execute offloaded code in a separate CPU process on the same host, simplifying debugging and avoiding the need for a GPU setup. The work involves creating a host-ISA offloaded code generation mode and a libgomp plugin to interface with the libgomp offloading machinery, managing the new process for the offloaded code. This approach can simplify debugging issues related to memory mapping or multi-device offloading, and potentially simulate GPU-like parallelism using CPU threads for a broader range of debugging scenarios. Please see the details in Git commit log of the attached patch. From 4e7fa1186f254a8cecb2527de01879bd8d910261 Mon Sep 17 00:00:00 2001 From: Georgii Burunsuzian Date: Fri, 4 Oct 2024 12:59:15 +0300 Subject: [PATCH] WIP: Initial submission of libgomp host_process plugin This commit adds the plugin-host_process.c implementation for the host_process plugin. The submission is marked as Work in Progress (WIP) due to several known limitations and incomplete features. Note that this submission does not include updates to the build and test harness, which will be submitted separately once the plugin is ready for upstream inclusion. Signed-off-by: Georgii Burunsuzian --- libgomp/plugin/plugin-host_process.c | 1066 ++++++++++++++++++++++++++ 1 file changed, 1066 insertions(+) create mode 100644 libgomp/plugin/plugin-host_process.c diff --git a/libgomp/plugin/plugin-host_process.c b/libgomp/plugin/plugin-host_process.c new file mode 100644 index 00000000000..a5420bf8908 --- /dev/null +++ b/libgomp/plugin/plugin-host_process.c @@ -0,0 +1,1066 @@ +/* Host process plugin */ + +#pragma GCC optimize ("O0") + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include "config.h" +#include "symcat.h" +#include "libgomp-plugin.h" +#include "oacc-plugin.h" +#include "gomp-constants.h" +#include "oacc-int.h" +#include + +#define MAX_DEVICES 1 +#define MAX_ALLOCATIONS 100 +#define ASYNC_QUEUE_SIZE 64 +//#define HOST_PROCESS_DEBUG(...) GOMP_PLUGIN_debug("HOST_PROCESS debug: ", __VA_ARGS__) + +const char * +GOMP_OFFLOAD_get_name (void) +{ + return "host_process"; +} + +unsigned int +GOMP_OFFLOAD_get_caps (void) +{ + return GOMP_OFFLOAD_CAP_NATIVE_EXEC | GOMP_OFFLOAD_CAP_OPENMP_400 | GOMP_OFFLOAD_CAP_OPENACC_200; +} + +int +GOMP_OFFLOAD_get_type (void) +{ + return OFFLOAD_TARGET_TYPE_HOST_PROCESS; +} + +int +GOMP_OFFLOAD_get_num_devices (unsigned int omp_requires_mask) +{ + return 1; +} + +bool +call_function_with_args(void *fn_ptr, void *vars) +{ + void (*fn) (void *) = fn_ptr; + fn(vars); + return true; +} + + +// struct to track allocations +typedef struct { + void* ptr; + size_t size; +} allocation; + + +// struct to store device information +// I thought pthread_mutex_t can be important if multiple threads +// will try to initialize, modify, or interact with the simulated device concurrently. +typedef struct { + bool initialized; + pthread_mutex_t lock; + pid_t process_id; + int socket_fd; + allocation allocations[MAX_ALLOCATIONS]; + void *image_memory; + size_t image_size; + struct goacc_asyncqueue *async_queues; + pthread_mutex_t async_queues_mutex; +} simulated_device; + +// Assuming i have an array of devices +simulated_device devices[MAX_DEVICES]; + + +bool +wrong_device_number(int n) +{ + if (n < 0 || n >= MAX_DEVICES) { + GOMP_PLUGIN_error ("wrong_device_number: Invalid device number %d\n", n); + return true; + } + return false; +} + +bool +check_device_validity(int n) +{ + if (wrong_device_number(n)) { + return false; + } + + simulated_device *device = &devices[n]; + if (!device->initialized || device->socket_fd < 0) { + GOMP_PLUGIN_error("check_device_validity: Device %d not initialized or socket not valid\n", n); + return false; + } + return true; +} + + +void +handle_alloc(int n, const char *command_details) +{ + simulated_device *device = &devices[n]; + size_t size; + int parsed = sscanf(command_details, "%zu", &size); + if (parsed != 1) { + GOMP_PLUGIN_error("handle_alloc: Failed to parse ALLOC command details\n"); + send(device->socket_fd, "ERROR", 5, 0); + return; + } + + void *ptr = malloc(size); + send(device->socket_fd, &ptr, sizeof(ptr), 0); + +// int index = -1; +// for (int i = 0; i < MAX_ALLOCATIONS; i++) { +// if (device->allocations[i].ptr == NULL) { +// index = i; +// break; +// } +// } +// +// void *ptr = NULL; +// if (index != -1) { +// ptr = malloc(size); +// if (ptr) { +// device->allocations[index].ptr = ptr; +// device->allocations[index].size = size; +// } +// } +} + + +void +handle_free(int n, const char *command_details) +{ + simulated_device *device = &devices[n]; + void *ptr; + + int parsed = sscanf(command_details, "%p", &ptr); + if (parsed != 1) { + GOMP_PLUGIN_error("handle_free: Failed to parse FREE command details\n"); + send(device->socket_fd, "ERROR", 5, 0); + return; + } + + free(ptr); + send(device->socket_fd, "OK", 2, 0); + + +//// If i am tracking allocations in 'allocations' array. +// for (int i = 0; i < MAX_ALLOCATIONS; i++) { +// if (device->allocations[i].ptr == ptr) { +// free(ptr); +// device->allocations[i].ptr = NULL; +// device->allocations[i].size = 0; +// break; +// } +// } +} + +void +handle_load_image(int n, char *command_details) +{ + + simulated_device *device = &devices[n]; + size_t size; + + int parsed = sscanf(command_details, "%zu", &size); + if (parsed != 1) { + GOMP_PLUGIN_error("handle_load_image: Failed to parse LOAD_IMAGE command details\n"); + send(device->socket_fd, "ERROR", 5, 0); + return; + } + size_t prefix_length = snprintf(NULL, 0, "%zu ", size); + void *image_memory = malloc(size); + if (!image_memory) { + GOMP_PLUGIN_error("handle_load_image: Failed to allocate memory for image\n"); + return; + } + image_memory = command_details + prefix_length; + + device->image_memory = image_memory; + device->image_size = size; + send(device->socket_fd, &image_memory, sizeof(image_memory), 0); +} + +void +handle_unload_image(int n) +{ + simulated_device *device = &devices[n]; + + if (device->image_memory == NULL) { + GOMP_PLUGIN_error("handle_unload_image: No image is currently loaded on device %d\n", n); + char response[64] = "ERROR, No image loaded"; + send(device->socket_fd, response, strlen(response), 0); + return; + } + + free(device->image_memory); + device->image_memory = NULL; + device->image_size = 0; + + GOMP_PLUGIN_debug(0, "handle_unload_image: Image successfully unloaded from device %d\n", n); + + char response[64] = "OK, Image unloaded"; + send(device->socket_fd, response, strlen(response), 0); +} + +void +handle_host2dev(int n, const char *command_details, const char *data_end) +{ + simulated_device *device = &devices[n]; + void *dst; + size_t size; + + int parsed = sscanf(command_details, "%p %zu", &dst, &size); + if (parsed != 2) { + GOMP_PLUGIN_error("handle_host2dev: Failed to parse HOST_TO_DEV command details\n"); + send(device->socket_fd, "ERROR", 5, 0); + return; + } + size_t prefix_length = snprintf(NULL, 0, "%p %zu ", dst, size); + memcpy(dst, command_details + prefix_length, size); + send(device->socket_fd, "OK", 2, 0); + +// const char *data_start = command_details; +// for (int i = 0; i < 2; i++) { +// data_start = strchr(data_start, ' '); +// if (!data_start) { +// GOMP_PLUGIN_error("Failed to locate data start in command\n"); +// send(device->socket_fd, "ERROR", 5, 0); +// return; +// } +// data_start++; +// } + + + +// const char *data = command_details + data_offset; +// +// char *data = malloc(size); +// if (!data) { +// GOMP_PLUGIN_error("Memory allocation failed for data transfer\n"); +// send(device->socket_fd, "ERROR", 5, 0); +// return; +// } +// +// int bytes_received = recv(device->socket_fd, data, size, 0); +// if (bytes_received < (int)size) { +// GOMP_PLUGIN_error("Failed to receive all data\n"); +// free(data); +// send(device->socket_fd, "ERROR", 5, 0); +// return; +// } +// free(data); + +} + +void +handle_dev2host(int n, const char *command_details) +{ + simulated_device *device = &devices[n]; + void *src; + size_t size; + + int parsed = sscanf(command_details, "%p %zu", &src, &size); + if (parsed != 2) { + GOMP_PLUGIN_error("handle_dev2host: Failed to parse DEV_TO_HOST command details"); + size_t error_indicator = 0; // Here i want to send a specific error code + send(device->socket_fd, &error_indicator, sizeof(size_t), 0); + return; + } + + // Can i assume here that src points to valid memory in this process ? + if (send(device->socket_fd, src, size, 0) < size) { + GOMP_PLUGIN_error("handle_dev2host: Failed to send all data to host"); + } +} + +void handle_dev2dev(int n, const char *command_details) { + simulated_device *device = &devices[n]; + void *dst, *src; + size_t size; + + int parsed = sscanf(command_details, "%p %p %zu", &dst, &src, &size); + if (parsed != 3) { + GOMP_PLUGIN_error("handle_dev2dev: Failed to parse DEV_TO_DEV command details"); + send(device->socket_fd, "ERROR", 5, 0); + return; + } + + memcpy(dst, src, size); + send(device->socket_fd, "OK", 2, 0); +} + + + +void +handle_call_function(int n, const char *command_details) +{ + simulated_device *device = &devices[n]; + void *fn_ptr; + void *vars; + + int parsed = sscanf(command_details, "%p %p", &fn_ptr, &vars); + if (parsed != 2) { + GOMP_PLUGIN_error("handle_call_function: Failed to parse CALL_FN command details\n"); + send(device->socket_fd, "ERROR", 5, 0); + return; + } + +// void (*actual_function_pointer)(int, int) = (void(*)(int, int))fn_ptr; +// actual_function_pointer(vars); + + if(call_function_with_args(fn_ptr, vars)) { + send(device->socket_fd, "OK", 2, 0); + } else { + send(device->socket_fd, "ERROR", 5, 0); + } +} + +void +child_process(int n) +{ + simulated_device *device = &devices[n]; + char *buffer = malloc(1 << 20); +// int pointer = 0; + + while (1) { + size_t message_length; + size_t bytes_received = recv(device->socket_fd, &message_length, sizeof(message_length), 0); + if (bytes_received < sizeof(message_length)) { + GOMP_PLUGIN_debug(0, "child_process: Failed to receive message length: 'bytes_received' : %zu, 'sizeof(message_length)' : %zu.\n", bytes_received, sizeof(message_length)); +// return; + } + + bytes_received = recv(device->socket_fd, buffer, message_length, 0); + if (bytes_received == 0) { // connection was closed by the other side + break; + } else if (bytes_received < 0) { + if (errno == EAGAIN || errno == EWOULDBLOCK) { + continue; // no data available right now, wait and try again + } else { + perror("recv failed"); + break; + } + } + + size_t remaining = message_length; + remaining -= bytes_received; + while (remaining > 0) { + bytes_received = recv(device->socket_fd, buffer + bytes_received, remaining, 0); +// error handling + remaining -= bytes_received; + } + + + buffer[bytes_received] = '\0'; + if (strncmp(buffer, "ALLOC", 5) == 0) { + handle_alloc(n, buffer + 6); + } else if (strncmp(buffer, "FREE", 4) == 0) { + handle_free(n, buffer + 5); + } else if (strncmp(buffer, "LOAD_IMAGE", 10) == 0) { + handle_load_image(n, buffer + 11); + } else if (strncmp(buffer, "UNLOAD_IMAGE", 12) == 0) { + handle_unload_image(n); + } else if (strncmp(buffer, "HOST_TO_DEV", 11) == 0) { + handle_host2dev(n, buffer + 12, &buffer[bytes_received]); + } else if (strncmp(buffer, "DEV_TO_HOST", 11) == 0) { + handle_dev2host(n, buffer + 12); + } else if (strncmp(buffer, "CALL_FN", 7) == 0) { + handle_call_function(n, buffer + 8); + } else if (strncmp(buffer, "DEV_TO_DEV", 10) == 0) { + handle_dev2dev(n, buffer + 11); +// } else if (strncmp(buffer, "OACC_CALL_FN", 12) == 0) { +// handle_call_function(n, buffer + 13); + } + } + close(device->socket_fd); +} + +bool +send_size_and_data(int n, char* command_name, char* command, size_t command_size) +{ + simulated_device *device = &devices[n]; + int bytes_sent = send(device->socket_fd, &command_size, sizeof(command_size), 0); + if (bytes_sent < 0) { + GOMP_PLUGIN_error("send_size_and_data: Failed to send command size\n"); + return false; + } + + bytes_sent = send(device->socket_fd, command, command_size, 0); + if (bytes_sent < 0) { + GOMP_PLUGIN_error("send_size_and_data: Failed to send %s command and data to device\n", command_name); + return false; + } + return true; +} + +bool +validate_socket_connection(int socket_fd) +{ + if (socket_fd < 0) { + GOMP_PLUGIN_error("validate_socket_connection: Invalid socket descriptor.\n"); + return false; + } + + int error = 0; + socklen_t errlen = sizeof(error); + if (getsockopt(socket_fd, SOL_SOCKET, SO_ERROR, &error, &errlen) != 0) { + GOMP_PLUGIN_error("validate_socket_connection: Socket operation failed.\n"); + return false; + } + + if (error != 0) { + GOMP_PLUGIN_error("validate_socket_connection: Socket error: %s\n", strerror(error)); + return false; + } + + return true; +} + + +bool +GOMP_OFFLOAD_init_device(int n) +{ + if (wrong_device_number(n)) { + return false; + } + + simulated_device *device = &devices[n]; + if (device->initialized) { + return true; + } + + // Initialize a mutex for this device + if (pthread_mutex_init(&device->lock, NULL) != 0) { + GOMP_PLUGIN_error("GOMP_OFFLOAD_init_device: Failed to initialize a mutex for simulated device %d\n", n); + return false; + } + + int sockets[2]; // Socket pair + if (socketpair(AF_UNIX, SOCK_STREAM, 0, sockets) != 0) { + GOMP_PLUGIN_error("GOMP_OFFLOAD_init_device: Failed to create socket pair\n"); + pthread_mutex_destroy(&device->lock); + return false; + } + + pid_t pid = fork(); + if (pid == -1) { + GOMP_PLUGIN_error("GOMP_OFFLOAD_init_device: Failed to fork a simulated device process\n"); + pthread_mutex_destroy(&device->lock); + close(sockets[0]); + close(sockets[1]); + return false; + } else if (pid == 0) { + close(sockets[0]); + device->socket_fd = sockets[1]; // here i save the parent's socket descriptor + device->initialized = true; + child_process(n); // function for child process + _exit(0); + } + + // Parent process + close(sockets[1]); + device->socket_fd = sockets[0]; // here i save the parent's socket descriptor + device->process_id = pid; + device->initialized = true; + GOMP_PLUGIN_debug(0, "GOMP_OFFLOAD_init_device: Simulated device %d initialized with process ID %d and socket FD %d\n", n, pid, sockets[0]); + + return true; +} + + +bool +GOMP_OFFLOAD_fini_device(int n) +{ + if (wrong_device_number(n)) { + return false; + } + + simulated_device *device = &devices[n]; + if (!device->initialized) { + return true; + } + + // Terminate the simulated device process + if (device->process_id != 0) { + if (kill(device->process_id, SIGTERM) == -1) { + GOMP_PLUGIN_error("GOMP_OFFLOAD_fini_device: Failed to terminate simulated device process\n"); + return false; + } + + // Wait for the process to terminate + int status; + if (waitpid(device->process_id, &status, 0) == -1) { + GOMP_PLUGIN_error("GOMP_OFFLOAD_fini_device: Failed to wait for simulated device process to terminate\n"); + return false; + } + } + + // destroy the mutex + if (pthread_mutex_destroy(&device->lock) != 0) { + GOMP_PLUGIN_error("GOMP_OFFLOAD_fini_device: Failed to destroy mutex for simulated device %d\n", n); + return false; + } + + // Current device is marked uninitialized + close(device->socket_fd); + device->initialized = false; + device->process_id = 0; + + return true; +} + + + +unsigned +GOMP_OFFLOAD_version (void) +{ + return GOMP_VERSION; +} + +typedef struct { + void *code; // Pointer to the compiled code + size_t code_size; // Size of the code + void **data; // Pointers to any relevant data + size_t data_size; // Size of the data +} host_process_image_desc; + +// Do i understand correctly that "Reverse offloading" refers to the scenario where +// functions that are normally executed on the device are instead executed on the host ? +// Will i need to handle indirect or reverse function calls ? +int +GOMP_OFFLOAD_load_image(int n, unsigned version, const void *target_data, + struct addr_pair **target_table, + uint64_t **rev_fn_table, + uint64_t *host_ind_fn_table) +{ +// if (GOMP_VERSION_DEV (version) != GOMP_VERSION_HOST_PROCESS) +// { +// GOMP_PLUGIN_error ("Offload data incompatible with HOST_PROCESS plugin" +// " (expected %u, received %u)", +// GOMP_VERSION_HOST_PROCESS, GOMP_VERSION_DEV (version)); +// return false; +// } + + if(!check_device_validity(n)) return false; + simulated_device *device = &devices[n]; + +// host_process_image_desc *img = (host_process_image_desc *)target_data; +// size_t total_size = img->code_size + img->data_size; + const size_t data_size = 0; + + int prefix_len = snprintf(NULL, 0, "LOAD_IMAGE %zu ", data_size); + size_t total_length = prefix_len + data_size; + + char *buffer = malloc(total_length); + if (!buffer) { + GOMP_PLUGIN_error("GOMP_OFFLOAD_load_image: Failed to allocate buffer for command and data\n"); + return false; + } + + int written = sprintf(buffer, "LOAD_IMAGE %zu ", data_size); + memcpy(buffer + written, target_data, data_size); + if(!send_size_and_data(n, "LOAD_IMAGE", buffer, total_length)){ + free(buffer); + return false; + } + free(buffer); + + void *response_address; + int recv_size = recv(device->socket_fd, &response_address, sizeof(response_address), 0); + if (recv_size < sizeof(response_address)) { + GOMP_PLUGIN_error("GOMP_OFFLOAD_load_image: Failed to receive response from child process\n"); + return -1; + } + + return 0; +} + +bool +GOMP_OFFLOAD_unload_image(int n, unsigned version, const void *target_data) +{ +// if (GOMP_VERSION_DEV (version) != GOMP_VERSION_HOST_PROCESS) +// { +// GOMP_PLUGIN_error ("Offload data incompatible with HOST_PROCESS plugin" +// " (expected %u, received %u)", +// GOMP_VERSION_HOST_PROCESS, GOMP_VERSION_DEV (version)); +// return false; +// } + + if(!check_device_validity(n)) return false; + simulated_device *device = &devices[n]; + + char command[256]; + size_t message_length = sprintf(command, "UNLOAD_IMAGE"); + if(!send_size_and_data(n, "UNLOAD_IMAGE", command, message_length)){ + return false; + } + + char response[64]; + int bytes_received = recv(device->socket_fd, response, sizeof(response), 0); + if (bytes_received <= 0) { + GOMP_PLUGIN_error("GOMP_OFFLOAD_unload_image: Failed to receive confirmation from device about unload operation\n"); + return false; + } + response[bytes_received] = '\0'; + GOMP_PLUGIN_debug(0, "GOMP_OFFLOAD_unload_image, received response: %s\n", response); + + return true; +} + +// What pointer should 'GOMP_OFFLOAD_alloc' return, if i am allocating in another address space ?.. +void * +GOMP_OFFLOAD_alloc(int n, size_t size) +{ + if(size == 0) return NULL; // as far as i understand, malloc(0) returns NULL + if(!check_device_validity(n)) return NULL; + simulated_device *device = &devices[n]; + +//// pthread_mutex_lock(&device->lock); +// if (!validate_socket_connection(device->socket_fd)) { +// GOMP_PLUGIN_error("Attempting to reconnect...\n"); +// // Reconnect or reinitialize the socket +// } + + char command[256]; + size_t message_length = sprintf(command, "ALLOC %zu", size); + if(!send_size_and_data(n, "ALLOC", command, message_length)){ + return NULL; + } + + // Seems like here i am expecting a response with the address of the allocated memory + void *ptr; + int bytes_received = recv(device->socket_fd, &ptr, sizeof(ptr), 0); + if (bytes_received < sizeof(ptr)) { + GOMP_PLUGIN_error("GOMP_OFFLOAD_alloc: Failed to receive allocation response or allocation failed\n"); +//// pthread_mutex_unlock(&device->lock); + return NULL; + } + +//// pthread_mutex_unlock(&device->lock); + return ptr; //// in process 'A' i am returning pointer from the address space of another process... +} + + +bool +GOMP_OFFLOAD_free(int n, void *ptr) +{ + if (ptr == NULL) return true; + if(!check_device_validity(n)) return false; + simulated_device *device = &devices[n]; + +// pthread_mutex_lock(&device->lock); +// bool found = false; +// for (int i = 0; i < MAX_ALLOCATIONS; i++) { +// if (device->allocations[i].ptr == ptr) { +// char command[256]; +// sprintf(command, "FREE %p", ptr); +// if (send(device->socket_fd, command, strlen(command), 0) < 0) { +// GOMP_PLUGIN_error("Failed to send free command\n"); +////// pthread_mutex_unlock(&device->lock); +// return false; +// } +// +// found = true; +// break; +// } +// } +//// pthread_mutex_unlock(&device->lock); +// if (!found) { +// GOMP_PLUGIN_error("Pointer not recognized or already freed for device %d\n", n); +// return true; +// } + + char command[256]; + size_t message_length = sprintf(command, "FREE %p", ptr); + if(!send_size_and_data(n, "FREE", command, message_length)){ + return false; + } + + char response[64]; + int bytes_received = recv(device->socket_fd, response, sizeof(response), 0); + if (bytes_received <= 0) { + GOMP_PLUGIN_error("GOMP_OFFLOAD_free: Failed to receive confirmation from device about free operation\n"); + return false; + } + response[bytes_received] = '\0'; + GOMP_PLUGIN_debug(0, "GOMP_OFFLOAD_free, received response: %s\n", response); + + return true; +} + + +//// THIS FUNCTION CHECKS THAT 'ptr' is in the memory that was allocated +//bool +//is_valid_memory(simulated_device *dev, void *ptr, size_t size) +//{ +// for (int i = 0; i < MAX_ALLOCATIONS; i++) { +// void *start = dev->allocations[i].ptr; +// size_t len = dev->allocations[i].size; +// if (ptr >= start && (char *)ptr + size <= (char *)start + len) { +// return true; +// } +// } +// return false; +//} + + +bool +GOMP_OFFLOAD_dev2host(int n, void *dst, const void *src, size_t size) +{ + if(size == 0) return true; + if(!check_device_validity(n)) return false; + simulated_device *device = &devices[n]; + + char command[256]; + size_t message_length = sprintf(command, "DEV_TO_HOST %p %zu", src, size); + if(!send_size_and_data(n, "DEV_TO_HOST", command, message_length)){ + return false; + } + + // Receive the data + int bytes_received = recv(device->socket_fd, dst, size, 0); + if (bytes_received < (int)size) { + GOMP_PLUGIN_error("GOMP_OFFLOAD_dev2host: Failed to receive all data from device\n"); + return false; + } + + return true; +} + + +bool +GOMP_OFFLOAD_host2dev(int n, void *dst, const void *src, size_t size) +{ + if(size == 0) return true; + if(!check_device_validity(n)) return false; + simulated_device *device = &devices[n]; + + int prefix_len = snprintf(NULL, 0, "HOST_TO_DEV %p %zu ", dst, size); + size_t total_length = prefix_len + size; + + char *buffer = malloc(total_length); + if (!buffer) { + GOMP_PLUGIN_error("GOMP_OFFLOAD_host2dev: Failed to allocate buffer for command and data\n"); + return false; + } + + // Write the command, including the offset to the data, into the buffer + int written = sprintf(buffer, "HOST_TO_DEV %p %zu ", dst, size); + memcpy(buffer + written, src, size); + + if(!send_size_and_data(n, "HOST_TO_DEV", buffer, total_length)){ + free(buffer); + return false; + } + free(buffer); + + char response[64]; + int bytes_received = recv(device->socket_fd, response, sizeof(response), 0); + if (bytes_received <= 0) { + GOMP_PLUGIN_error("GOMP_OFFLOAD_host2dev: Failed to receive confirmation from device about successful copy from host to device\n"); + return false; + } + response[bytes_received] = '\0'; + + GOMP_PLUGIN_debug(0, "GOMP_OFFLOAD_host2dev, received response: %s\n", response); + if (strncmp(response, "OK", 2) != 0) { + GOMP_PLUGIN_error("GOMP_OFFLOAD_host2dev: Device reported an error during data transfer\n"); + return false; + } + + return true; +} + +void +GOMP_OFFLOAD_run (int n, void *fn_ptr, void *vars, void **args) +{ + if(fn_ptr == NULL) { + GOMP_PLUGIN_error("GOMP_OFFLOAD_run: function pointer is NULL\n"); + return; + } + if(!check_device_validity(n)) return; + simulated_device *device = &devices[n]; + + char command[256]; + size_t message_length = sprintf(command, "CALL_FN %p %p", fn_ptr, vars); + if(!send_size_and_data(n, "CALL_FN", command, message_length)){ + return; + } + + char response[64]; + int bytes_received = recv(device->socket_fd, response, sizeof(response), 0); + if (bytes_received <= 0) { + GOMP_PLUGIN_error("GOMP_OFFLOAD_run: Failed to receive confirmation from device about successful function call\n"); + return; + } + response[bytes_received] = '\0'; + GOMP_PLUGIN_debug(0, "GOMP_OFFLOAD_run, received response: %s\n", response); +} + + + +bool +GOMP_OFFLOAD_dev2dev (int n, void *dst, const void *src, size_t size) +{ + if(size == 0) return true; + if(!check_device_validity(n)) return false; + simulated_device *device = &devices[n]; + + char command[256]; + size_t message_length = sprintf(command, "DEV_TO_DEV %p %p %zu", dst, src, size); + if(!send_size_and_data(n, "DEV_TO_DEV", command, message_length)){ + return false; + } + + char response[64]; + int bytes_received = recv(device->socket_fd, response, sizeof(response), 0); + if (bytes_received <= 0) { + GOMP_PLUGIN_error("GOMP_OFFLOAD_dev2dev: Failed to receive confirmation from device about successful copy\n"); + return false; + } + response[bytes_received] = '\0'; + GOMP_PLUGIN_debug(0, "GOMP_OFFLOAD_dev2dev, received response: %s\n", response); + + if (strncmp(response, "OK", 2) != 0) { + GOMP_PLUGIN_error("GOMP_OFFLOAD_dev2dev: Device reported an error during data copy\n"); + return false; + } + + return true; +} + + + + + + + +//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// +//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// +//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// +//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// +//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// + + + + + +void +GOMP_OFFLOAD_openacc_destroy_thread_data (void *data) +{ + free (data); +} + +void +GOMP_OFFLOAD_openacc_exec (void (*fn) (void *), + size_t mapnum __attribute__ ((unused)), + void **hostaddrs, + void **devaddrs __attribute__ ((unused)), + unsigned *dims __attribute__ ((unused)), + void *targ_mem_desc __attribute__ ((unused))) +{ + GOMP_PLUGIN_debug(0, "Beginning of GOMP_OFFLOAD_openacc_exec function\n"); + GOMP_OFFLOAD_run(0, fn, devaddrs, NULL); + GOMP_PLUGIN_debug(0, "End of GOMP_OFFLOAD_openacc_exec function\n"); +} + +void +GOMP_OFFLOAD_openacc_async_exec (void (*fn) (void *), + size_t mapnum __attribute__ ((unused)), + void **hostaddrs, + void **devaddrs __attribute__ ((unused)), + unsigned *dims __attribute__ ((unused)), + void *targ_mem_desc __attribute__ ((unused)), + struct goacc_asyncqueue *aq __attribute__ ((unused))) +{ + GOMP_PLUGIN_debug(0, "Beginning of GOMP_OFFLOAD_openacc_async_exec function\n"); + GOMP_OFFLOAD_run(0, fn, devaddrs, NULL); + GOMP_PLUGIN_debug(0, "End of GOMP_OFFLOAD_openacc_async_exec function\n"); +// GOMP_PLUGIN_fatal ("%s", __FUNCTION__); +} + + +typedef struct { + void (*function)(void* data); + void *data; +} queue_task; + + +struct goacc_asyncqueue +{ + simulated_device *device; +// hsa_queue_t *hsa_queue; + + pthread_t thread_drain_queue; + pthread_mutex_t mutex; + pthread_cond_t queue_cond_in; + pthread_cond_t queue_cond_out; + queue_task queue[ASYNC_QUEUE_SIZE]; // queue_task should be universal, but now it is only for copy operations. + int queue_first; + int queue_n; + int drain_queue_stop; + + int id; + struct goacc_asyncqueue *prev; + struct goacc_asyncqueue *next; +}; + + +bool +GOMP_OFFLOAD_openacc_async_host2dev(int device, void *dst, const void *src, size_t n, struct goacc_asyncqueue *aq) +{ + GOMP_PLUGIN_debug(0, "Beginning of GOMP_OFFLOAD_openacc_async_host2dev function\n"); + bool result = GOMP_OFFLOAD_host2dev(device, dst, src, n); + GOMP_PLUGIN_debug(0, "End of GOMP_OFFLOAD_openacc_async_host2dev function\n"); + return result; +// GOMP_PLUGIN_fatal ("%s", __FUNCTION__); +// return false; +} + + +bool +GOMP_OFFLOAD_openacc_async_dev2host(int device, void *dst, const void *src, size_t n, struct goacc_asyncqueue *aq) +{ + GOMP_PLUGIN_debug(0, "Beginning of GOMP_OFFLOAD_openacc_async_dev2host function\n"); + bool result = GOMP_OFFLOAD_dev2host(device, dst, src, n); + GOMP_PLUGIN_debug(0, "End of GOMP_OFFLOAD_openacc_async_dev2host function\n"); + return result; +// GOMP_PLUGIN_fatal ("%s", __FUNCTION__); +// return false; +} + +struct goacc_asyncqueue * +GOMP_OFFLOAD_openacc_async_construct(int n) +{ + return (struct goacc_asyncqueue *) -1; +// GOMP_PLUGIN_fatal ("%s", __FUNCTION__); +// return NULL; +} + + +bool +GOMP_OFFLOAD_openacc_async_destruct(struct goacc_asyncqueue *aq) +{ + return true; +// GOMP_PLUGIN_fatal ("%s", __FUNCTION__); +// return false; +} + + + +/// OPTIONAL +//extern int GOMP_OFFLOAD_memcpy2d (int, int, size_t, size_t, +// void*, size_t, size_t, size_t, +// const void*, size_t, size_t, size_t); +//extern int GOMP_OFFLOAD_memcpy3d (int, int, size_t, size_t, size_t, void *, +// size_t, size_t, size_t, size_t, size_t, +// const void *, size_t, size_t, size_t, size_t, +// size_t); + + +bool +GOMP_OFFLOAD_can_run (void *fn_ptr) +{ + return true; +} + + +void +GOMP_OFFLOAD_async_run (int device, void *tgt_fn, void *tgt_vars, + void **args, void *async_data) +{ + void (*fn)(void *) = (void (*)(void *)) tgt_fn; + fn (tgt_vars); +// GOMP_PLUGIN_fatal ("%s", __FUNCTION__); +} + +void * +GOMP_OFFLOAD_openacc_create_thread_data (int ord __attribute__((unused))) +{ + return NULL; +} + +int +GOMP_OFFLOAD_openacc_async_test (struct goacc_asyncqueue *aq) +{ + return 1; +// GOMP_PLUGIN_fatal ("%s", __FUNCTION__); +// return 0; +} + +bool +GOMP_OFFLOAD_openacc_async_synchronize (struct goacc_asyncqueue *aq) +{ + return true; +// GOMP_PLUGIN_fatal ("%s", __FUNCTION__); +// return false; +} + +bool +GOMP_OFFLOAD_openacc_async_serialize (struct goacc_asyncqueue *aq1, + struct goacc_asyncqueue *aq2) +{ + return true; +// GOMP_PLUGIN_fatal ("%s", __FUNCTION__); +// return false; +} + +void +GOMP_OFFLOAD_openacc_async_queue_callback (struct goacc_asyncqueue *aq, + void (*fn) (void *), void *data) +{ +// GOMP_PLUGIN_fatal ("%s", __FUNCTION__); +} + + +//extern void *GOMP_OFFLOAD_openacc_cuda_get_current_device (void); +//extern void *GOMP_OFFLOAD_openacc_cuda_get_current_context (void); +//extern void *GOMP_OFFLOAD_openacc_cuda_get_stream (struct goacc_asyncqueue *); +//extern int GOMP_OFFLOAD_openacc_cuda_set_stream (struct goacc_asyncqueue *, +// void *); + + +union goacc_property_value +GOMP_OFFLOAD_openacc_get_property (int n, enum goacc_property prop) +{ + union goacc_property_value nullval = { .val = 0 }; + + if (n >= GOMP_OFFLOAD_get_num_devices (0)) + return nullval; + + switch (prop) + { + case GOACC_PROPERTY_NAME: + return (union goacc_property_value) { .ptr = "GOMP" }; + case GOACC_PROPERTY_VENDOR: + return (union goacc_property_value) { .ptr = "GNU" }; + case GOACC_PROPERTY_DRIVER: + return (union goacc_property_value) { .ptr = VERSION }; + case GOACC_PROPERTY_MEMORY: + case GOACC_PROPERTY_FREE_MEMORY: + default: + return nullval; + } +} -- 2.43.0