kc3-lang/angle/src/libANGLE/renderer/metal/mtl_library_cache.h

• Show log

  Commit

• Hash : d4083c79
  Author :
  Date : 2025-01-31T15:06:36
  

  Metal: Make ObjCPtr available to all modules
  
  Move ObjCPtr implementation to src/common/ObjCPtr.h,
  angle::ObjCPtr so that it's available also outside libANGLE.
  
  Bug: angleproject:393263506
  Change-Id: I8ecd5632c7ae33ef4b409fc820fad229e91bc0ab
  Reviewed-on: https://chromium-review.googlesource.com/c/angle/angle/+/6219318
  Auto-Submit: Kimmo Kinnunen <kkinnunen@apple.com>
  Reviewed-by: Geoff Lang <geofflang@chromium.org>
  Commit-Queue: Kimmo Kinnunen <kkinnunen@apple.com>
  

Download

• src/libANGLE/renderer/metal/mtl_library_cache.h

• //
  // Copyright 2023 The ANGLE Project Authors. All rights reserved.
  // Use of this source code is governed by a BSD-style license that can be
  // found in the LICENSE file.
  //
  // mtl_library_cache.h:
  //    Defines classes for caching of mtl libraries
  //
  
  #ifndef LIBANGLE_RENDERER_METAL_MTL_LIBRARY_CACHE_H_
  #define LIBANGLE_RENDERER_METAL_MTL_LIBRARY_CACHE_H_
  
  #include "libANGLE/renderer/metal/mtl_utils.h"
  
  #include <type_traits>
  
  namespace rx
  {
  namespace mtl
  {
  
  class LibraryCache : angle::NonCopyable
  {
    public:
      LibraryCache();
  
      angle::ObjCPtr<id<MTLLibrary>> get(const std::shared_ptr<const std::string> &source,
                                         const std::map<std::string, std::string> &macros,
                                         bool disableFastMath,
                                         bool usesInvariance);
      angle::ObjCPtr<id<MTLLibrary>> getOrCompileShaderLibrary(
          DisplayMtl *displayMtl,
          const std::shared_ptr<const std::string> &source,
          const std::map<std::string, std::string> &macros,
          bool disableFastMath,
          bool usesInvariance,
          angle::ObjCPtr<NSError> *errorOut);
  
    private:
      struct LibraryKey
      {
          LibraryKey() = default;
          LibraryKey(const std::shared_ptr<const std::string> &source,
                     const std::map<std::string, std::string> &macros,
                     bool disableFastMath,
                     bool usesInvariance);
  
          std::shared_ptr<const std::string> source;
          std::map<std::string, std::string> macros;
          bool disableFastMath;
          bool usesInvariance;
  
          bool operator==(const LibraryKey &other) const;
      };
  
      struct LibraryKeyHasher
      {
          // Hash function
          size_t operator()(const LibraryKey &k) const;
  
          // Comparison
          bool operator()(const LibraryKey &a, const LibraryKey &b) const;
      };
  
      struct LibraryCacheEntry : angle::NonCopyable
      {
          LibraryCacheEntry() = default;
          ~LibraryCacheEntry();
          LibraryCacheEntry(LibraryCacheEntry &&moveFrom);
  
          // library can only go from the null -> not null state. It is safe to check if the library
          // already exists without locking.
          angle::ObjCPtr<id<MTLLibrary>> library;
  
          // Lock for this specific library to avoid multiple threads compiling the same shader at
          // once.
          std::mutex lock;
      };
  
      LibraryCacheEntry &getCacheEntry(LibraryKey &&key);
  
      static constexpr unsigned int kMaxCachedLibraries = 128;
  
      // The cache tries to clean up this many states at once.
      static constexpr unsigned int kGCLimit = 32;
  
      // Lock for searching and adding new entries to the cache
      std::mutex mCacheLock;
  
      using CacheMap = angle::base::HashingMRUCache<LibraryKey, LibraryCacheEntry, LibraryKeyHasher>;
      CacheMap mCache;
  };
  
  }  // namespace mtl
  }  // namespace rx
  
  #endif  // LIBANGLE_RENDERER_METAL_MTL_LIBRARY_CACHE_H_
  

Download