@@ -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