Skip to content

Commit 39fa89c

Browse files
gcarranza-1copybara-github
authored andcommitted
Wrap LiteRtEnvOptionTag into C++ enum EnvironmentOptions::Tag.
LiteRT-PiperOrigin-RevId: 826522913
1 parent 8d0ef65 commit 39fa89c

File tree

1 file changed

+35
-2
lines changed

1 file changed

+35
-2
lines changed

litert/cc/litert_environment_options.h

Lines changed: 35 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,42 @@ class EnvironmentOptions
3535
explicit EnvironmentOptions(LiteRtEnvironmentOptions env)
3636
: NonOwnedHandle(env) {}
3737

38-
using OptionTag = LiteRtEnvOptionTag;
38+
enum class Tag : int {
39+
kCompilerPluginLibraryDir = kLiteRtEnvOptionTagCompilerPluginLibraryDir,
40+
kDispatchLibraryDir = kLiteRtEnvOptionTagDispatchLibraryDir,
41+
kOpenClDeviceId = kLiteRtEnvOptionTagOpenClDeviceId,
42+
kOpenClPlatformId = kLiteRtEnvOptionTagOpenClPlatformId,
43+
kOpenClContext = kLiteRtEnvOptionTagOpenClContext,
44+
kOpenClCommandQueue = kLiteRtEnvOptionTagOpenClCommandQueue,
45+
kEglDisplay = kLiteRtEnvOptionTagEglDisplay,
46+
kEglContext = kLiteRtEnvOptionTagEglContext,
47+
kWebGpuDevice = kLiteRtEnvOptionTagWebGpuDevice,
48+
kWebGpuQueue = kLiteRtEnvOptionTagWebGpuQueue,
49+
kMetalDevice = kLiteRtEnvOptionTagMetalDevice,
50+
kMetalCommandQueue = kLiteRtEnvOptionTagMetalCommandQueue,
51+
// WARNING: Vulkan support is experimental.
52+
kVulkanEnvironment = kLiteRtEnvOptionTagVulkanEnvironment,
53+
kVulkanCommandPool = kLiteRtEnvOptionTagVulkanCommandPool,
54+
kCallbackOnGpuEnvDestroy = kLiteRtEnvOptionTagCallbackOnGpuEnvDestroy,
55+
kCallbackUserDataOnGpuEnvDestroy =
56+
kLiteRtEnvOptionTagCallbackUserDataOnGpuEnvDestroy,
57+
kMagicNumberConfigs = kLiteRtEnvOptionTagMagicNumberConfigs,
58+
kMagicNumberVerifications = kLiteRtEnvOptionTagMagicNumberVerifications,
59+
kCompilerCacheDir = kLiteRtEnvOptionTagCompilerCacheDir,
60+
// Singleton ML Drift WebGPU/Dawn instance required for shared libraries not
61+
// to create their own instances.
62+
kWebGpuInstance = kLiteRtEnvOptionTagWebGpuInstance,
63+
// Dawn procedure table pointer for shared libraries to populate their
64+
// tables with the shared procedures instead of their own procedures.
65+
kWebGpuProcs = kLiteRtEnvOptionTagWebGpuProcs,
66+
};
3967

40-
Expected<LiteRtVariant> GetOption(OptionTag tag) const {
68+
Expected<LiteRtVariant> GetOption(Tag tag) const {
69+
return GetOption(static_cast<LiteRtEnvOptionTag>(tag));
70+
}
71+
72+
[[deprecated("Use GetOption(Tag) instead.")]]
73+
Expected<LiteRtVariant> GetOption(LiteRtEnvOptionTag tag) const {
4174
if (Get() == nullptr) {
4275
return Error(kLiteRtStatusErrorInvalidArgument,
4376
"Environment options are null");

0 commit comments

Comments
 (0)