| // Copyright 2024 The Chromium Authors |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef SERVICES_WEBNN_WEBNN_SWITCHES_H_ |
| #define SERVICES_WEBNN_WEBNN_SWITCHES_H_ |
| |
| #include "base/containers/span.h" |
| #include "build/build_config.h" |
| #include "services/webnn/buildflags.h" |
| |
| namespace switches { |
| |
| #if BUILDFLAG(IS_MAC) |
| // Copy the generated Core ML model to the folder specified |
| // by --webnn-coreml-dump-model, note: folder needs to be accessible from |
| // the GPU sandbox or use --no-sandbox. |
| // Usage: --no-sandbox --webnn-coreml-dump-model=/tmp/CoreMLModels |
| inline constexpr char kWebNNCoreMlDumpModel[] = "webnn-coreml-dump-model"; |
| #endif // BUILDFLAG(IS_MAC) |
| |
| #if BUILDFLAG(WEBNN_USE_TFLITE) |
| // Save the generated TFLite model file to the folder specified by |
| // --webnn-tflite-dump-model. Note, the folder needs to be accessible from the |
| // GPU process sandbox or --no-sandbox must be used. |
| // Usage: --no-sandbox --webnn-tflite-dump-model=/tmp/tflite_models |
| inline constexpr char kWebNNTfliteDumpModel[] = "webnn-tflite-dump-model"; |
| #endif // BUILDFLAG(WEBNN_USE_TFLITE) |
| |
| #if BUILDFLAG(IS_WIN) |
| // Configure the logging severity level of ONNX Runtime. |
| // Usage: --webnn-ort-logging-level=VERBOSE |
| // Other severity levels could be "INFO", "WARNING", "ERROR" (default), and |
| // "FATAL". |
| inline constexpr char kWebNNOrtLoggingLevel[] = "webnn-ort-logging-level"; |
| // Set the folder specified by --webnn-ort-dump-model for ONNX Runtime to save |
| // optimized ONNX model after graph level transformations. |
| // Usage: --no-sandbox --webnn-ort-dump-model=/tmp/ort_models |
| inline constexpr char kWebNNOrtDumpModel[] = "webnn-ort-dump-model"; |
| // Force onnxruntime.dll to be loaded from a location specified by the switch |
| // for testing development ORT build. This switch is not to be used in shipping |
| // scenarios and is ignored by default. |
| // Usage: --webnn-ort-library-path-for-testing="C:\Program Files\ONNXRuntime" |
| // --allow-third-party-modules |
| inline constexpr char kWebNNOrtLibraryPathForTesting[] = |
| "webnn-ort-library-path-for-testing"; |
| // Force ORT EP libraries to be loaded from a location specified by the switch |
| // for testing development EP build. This switch is not to be used in shipping |
| // scenarios and is ignored by default. |
| // Usage: --webnn-ort-ep-library-path-for-testing="C:\Program |
| // Files\ONNXRuntime-EP" --allow-third-party-modules |
| inline constexpr char kWebNNOrtEpLibraryPathForTesting[] = |
| "webnn-ort-ep-library-path-for-testing"; |
| |
| // Configure the graph optimization level of ONNX Runtime. |
| // Usage: --webnn-ort-graph-optimization-level=DISABLE_ALL |
| // Other levels could be "BASIC", "EXTENDED" and "ALL". |
| inline constexpr char kWebNNOrtGraphOptimizationLevel[] = |
| "webnn-ort-graph-optimization-level"; |
| |
| // This switch allows us to collect ORT profile data for performance analysis. |
| // The profile data file is generated in Chrome's folder with a fixed naming |
| // format "prefix_date_time.json". The prefix can be provided by user or use |
| // "WebNNOrtProfile" as default. |
| // Usage: --no-sandbox --webnn-ort-enable-profiling="WebNNOrtOvCpuProfile" |
| inline constexpr char kWebNNOrtEnableProfiling[] = "webnn-ort-enable-profiling"; |
| #endif // BUILDFLAG(IS_WIN) |
| |
| extern base::span<const char* const> GetWebNNSwitchesCopiedFromGpuProcessHost(); |
| // extern const base::span<const char* const> |
| // kWebNNSwitchesCopiedFromGpuProcessHost; |
| |
| } // namespace switches |
| |
| #endif // SERVICES_WEBNN_WEBNN_SWITCHES_H_ |
