iree_hal.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_TARGET_IREE_HAL_H_
#define LOOMC_TARGET_IREE_HAL_H_
 
#include <stdbool.h>
 
#include "iree/hal/api.h"
#include "loomc/module.h"
#include "loomc/sanitizer.h"
#include "loomc/target.h"

 
#ifdef __cplusplus
extern "C" {
#endif
 
typedef struct loomc_iree_hal_target_options_t loomc_iree_hal_target_options_t;

typedef struct loomc_iree_hal_target_provider_t
    loomc_iree_hal_target_provider_t;

LOOMC_API_EXPORT loomc_status_t loomc_iree_hal_module_query_runtime_features(
    const loomc_module_t* module,
    const loomc_sanitizer_options_t* sanitizer_options,
    const loomc_string_view_t* root_symbols,
    loomc_host_size_t root_symbol_count,
    iree_hal_device_runtime_feature_flags_t* out_runtime_features);

typedef struct loomc_iree_hal_target_selection_t {
  loomc_target_profile_t* target_profile;

  const iree_hal_executable_target_t* executable_target;
} loomc_iree_hal_target_selection_t;

typedef loomc_status_t(LOOMC_API_PTR* loomc_iree_hal_target_provider_fn_t)(
    void* user_data, loomc_target_environment_t* target_environment,
    const loomc_iree_hal_target_options_t* options, loomc_allocator_t allocator,
    bool* out_supported, loomc_iree_hal_target_selection_t* out_selection,
    loomc_result_t** out_result);

struct loomc_iree_hal_target_provider_t {
  loomc_string_view_t name;

  void* user_data;

  loomc_iree_hal_target_provider_fn_t select_target;
};

struct loomc_iree_hal_target_options_t {
  loomc_structure_type_t type;

  loomc_host_size_t structure_size;

  const void* next;

  loomc_string_view_t identifier;

  iree_hal_device_t* device;

  iree_hal_physical_device_affinity_t physical_device_affinity;

  loomc_target_profile_t* target_profile;

  const loomc_iree_hal_target_provider_t* const* providers;

  loomc_host_size_t provider_count;
};

LOOMC_API_EXPORT loomc_status_t loomc_target_select_iree_hal(
    loomc_target_environment_t* target_environment,
    const loomc_iree_hal_target_options_t* options, loomc_allocator_t allocator,
    loomc_iree_hal_target_selection_t* out_selection,
    loomc_result_t** out_result);
 
#ifdef __cplusplus
}  // extern "C"
#endif
 
#endif  // LOOMC_TARGET_IREE_HAL_H_