You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

option.cpp 1.4 kB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. // Tencent is pleased to support the open source community by making ncnn available.
  2. //
  3. // Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
  4. //
  5. // Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
  6. // in compliance with the License. You may obtain a copy of the License at
  7. //
  8. // https://opensource.org/licenses/BSD-3-Clause
  9. //
  10. // Unless required by applicable law or agreed to in writing, software distributed
  11. // under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
  12. // CONDITIONS OF ANY KIND, either express or implied. See the License for the
  13. // specific language governing permissions and limitations under the License.
  14. #include "option.h"
  15. #include "cpu.h"
  16. namespace ncnn {
  17. Option::Option()
  18. {
  19. lightmode = true;
  20. num_threads = get_cpu_count();
  21. blob_allocator = 0;
  22. workspace_allocator = 0;
  23. #if NCNN_VULKAN
  24. blob_vkallocator = 0;
  25. workspace_vkallocator = 0;
  26. staging_vkallocator = 0;
  27. #endif // NCNN_VULKAN
  28. use_winograd_convolution = true;
  29. use_sgemm_convolution = true;
  30. use_int8_inference = true;
  31. use_vulkan_compute = false;// TODO enable me
  32. use_fp16_packed = false;// TODO enable me
  33. use_fp16_storage = true;
  34. use_fp16_arithmetic = false;
  35. use_int8_storage = true;
  36. use_int8_arithmetic = false;
  37. // sanitize
  38. if (num_threads <= 0)
  39. num_threads = 1;
  40. }
  41. } // namespace ncnn