gni 0.1.4

Library to create a GPU Node ID
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

// Copyright (c) Meta Platforms, Inc. and affiliates.
// All rights reserved.

// This source code is licensed under the license found in the
// LICENSE file in the root directory of this source tree.

#include "GNI.h"
#include <stdlib.h>
#include <string.h>

extern char* rust_get_gpu_node_id();
extern void rust_free_ptr(char*);

const char* c_get_gpu_node_id() {
    char* rust_string = rust_get_gpu_node_id();
    if (!rust_string) {
        return NULL;
    }
    // Copy the string and free the original string.
    // source: https://en.cppreference.com/w/c/experimental/dynamic/strdup
    const char* result = strdup(rust_string);
    rust_free_ptr(rust_string);

    // Caller is responsible for freeing the result.
    return result;
}