benchmark/nixlbench: POSIX backend – LINUXAIO API

This patch adds support for a POSIX LINUXAIO API using the Linux AIO
plugin.

Signed-off-by: Anton Nayshtut <[email protected]>

Author: anton-nayshtut

benchmark/kvbench/commands/args.py

@@ -170,7 +170,7 @@ def nixl_bench_args(func):
     func = click.option(
         "--posix_api_type",
         type=str,
-        help="API type for POSIX operations [AIO, URING] (only used with POSIX backend",
+        help="API type for POSIX operations [AIO, URING, LINUXAIO] (only used with POSIX backend",
     )(func)
     func = click.option(
         "--enable_vmm",

benchmark/nixlbench/README.md

@@ -465,7 +465,7 @@ sudo systemctl start etcd && sudo systemctl enable etcd
 
 **POSIX Backend:**
 ```
---posix_api_type TYPE      # API type for POSIX operations [AIO, URING] (default: AIO)
+--posix_api_type TYPE      # API type for POSIX operations [AIO, URING, LINUXAIO] (default: AIO)
 ```
 
 **GPUNETIO Backend:**

benchmark/nixlbench/src/utils/utils.cpp

@@ -102,9 +102,10 @@ DEFINE_string(etcd_endpoints,
               "ETCD server endpoints for communication (optional for storage backends)");
 
 // POSIX options - only used when backend is POSIX
-DEFINE_string (posix_api_type,
-               XFERBENCH_POSIX_API_AIO,
-               "API type for POSIX operations [AIO, URING] (only used with POSIX backend)");
+DEFINE_string(
+    posix_api_type,
+    XFERBENCH_POSIX_API_AIO,
+    "API type for POSIX operations [AIO, URING, LINUXAIO] (only used with POSIX backend)");
 
 // DOCA GPUNetIO options - only used when backend is DOCA GPUNetIO
 DEFINE_string(gpunetio_device_list, "0", "Comma-separated GPU CUDA device id to use for \
@@ -211,9 +212,10 @@ xferBenchConfig::loadFromFlags() {
 
             // Validate POSIX API type
             if (posix_api_type != XFERBENCH_POSIX_API_AIO &&
-                posix_api_type != XFERBENCH_POSIX_API_URING) {
+                posix_api_type != XFERBENCH_POSIX_API_URING &&
+                posix_api_type != XFERBENCH_POSIX_API_LINUXAIO) {
                 std::cerr << "Invalid POSIX API type: " << posix_api_type
-                          << ". Must be one of [AIO, URING]" << std::endl;
+                          << ". Must be one of [AIO, URING, LINUXAIO]" << std::endl;
                 return -1;
             }
         }
@@ -412,7 +414,7 @@ xferBenchConfig::printConfig() {
 
         // Print POSIX options if backend is POSIX
         if (backend == XFERBENCH_BACKEND_POSIX) {
-            printOption ("POSIX API type (--posix_api_type=[AIO,URING])", posix_api_type);
+            printOption("POSIX API type (--posix_api_type=[AIO,URING,LINUXAIO])", posix_api_type);
         }
 
         // Print OBJ options if backend is OBJ

benchmark/nixlbench/src/utils/utils.h

@@ -79,6 +79,7 @@
 // POSIX API types
 #define XFERBENCH_POSIX_API_AIO "AIO"
 #define XFERBENCH_POSIX_API_URING "URING"
+#define XFERBENCH_POSIX_API_LINUXAIO "LINUXAIO"
 
 // OBJ S3 scheme types
 #define XFERBENCH_OBJ_SCHEME_HTTP "http"

benchmark/nixlbench/src/worker/nixl/nixl_worker.cpp

@@ -162,9 +162,15 @@ xferBenchNixlWorker::xferBenchNixlWorker(int *argc, char ***argv, std::vector<st
         if (xferBenchConfig::posix_api_type == XFERBENCH_POSIX_API_AIO) {
             backend_params["use_aio"] = "true";
             backend_params["use_uring"] = "false";
+            backend_params["use_linux_aio"] = "false";
         } else if (xferBenchConfig::posix_api_type == XFERBENCH_POSIX_API_URING) {
             backend_params["use_aio"] = "false";
             backend_params["use_uring"] = "true";
+            backend_params["use_linux_aio"] = "false";
+        } else if (xferBenchConfig::posix_api_type == XFERBENCH_POSIX_API_LINUXAIO) {
+            backend_params["use_aio"] = "false";
+            backend_params["use_uring"] = "false";
+            backend_params["use_linux_aio"] = "true";
         }
         std::cout << "POSIX backend with API type: " << xferBenchConfig::posix_api_type
                   << std::endl;