launch_config.h

Go to the documentation of this file.

// Copyright 2026 The IREE Authors
//
// Licensed under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 
#ifndef LOOMC_LAUNCH_CONFIG_H_
#define LOOMC_LAUNCH_CONFIG_H_
 
#include <stdbool.h>
#include <stdint.h>
 
#include "loomc/artifact.h"

 
#ifdef __cplusplus
extern "C" {
#endif

typedef struct loomc_launch_config_program_t loomc_launch_config_program_t;

typedef struct loomc_launch_config_function_t {
  uint64_t value;
} loomc_launch_config_function_t;

#define LOOMC_LAUNCH_CONFIG_FUNCTION_INVALID_VALUE UINT64_MAX

static inline loomc_launch_config_function_t
loomc_launch_config_function_invalid(void) {
  loomc_launch_config_function_t function = {
      LOOMC_LAUNCH_CONFIG_FUNCTION_INVALID_VALUE};
  return function;
}

static inline bool loomc_launch_config_function_is_valid(
    loomc_launch_config_function_t function) {
  return function.value != LOOMC_LAUNCH_CONFIG_FUNCTION_INVALID_VALUE;
}

typedef struct loomc_launch_config_t {
  loomc_structure_type_t type;

  loomc_host_size_t structure_size;

  void* next;

  loomc_dimension3_t workgroup_count;

  loomc_dimension3_t workgroup_size;

  loomc_dimension3_t workgroup_cluster_size;

  uint32_t subgroup_size;

  uint64_t workgroup_storage_bytes;
} loomc_launch_config_t;

LOOMC_API_EXPORT loomc_status_t loomc_launch_config_program_load(
    const loomc_artifact_t* artifact, loomc_allocator_t allocator,
    loomc_launch_config_program_t** out_program);

LOOMC_API_EXPORT void loomc_launch_config_program_retain(
    loomc_launch_config_program_t* program);

LOOMC_API_EXPORT void loomc_launch_config_program_release(
    loomc_launch_config_program_t* program);

LOOMC_API_EXPORT loomc_status_t loomc_launch_config_program_lookup_function(
    const loomc_launch_config_program_t* program,
    loomc_string_view_t export_name,
    loomc_launch_config_function_t* out_function);

LOOMC_API_EXPORT loomc_status_t
loomc_launch_config_program_invoke(loomc_launch_config_program_t* program,
                                   loomc_launch_config_function_t function,
                                   const uint64_t* workload_argument_bits,
                                   loomc_host_size_t workload_argument_count,
                                   loomc_launch_config_t* out_config);
 
#ifdef __cplusplus
}  // extern "C"
#endif
 
#endif  // LOOMC_LAUNCH_CONFIG_H_