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

• Show log

  Commit

• Hash : 75254b98
  Author :
  Date : 2023-06-22T15:06:29
  

  Metal: Cache compute pipelines with render pipelines.
  
  Support caching of compute pipelines in the same LRU cache as render
  pipelines in mtl::PipelineCache.
  
  Bug: chromium:1329376
  Change-Id: I93bbfadb8f5c1461144f1c222362c174402cced1
  Reviewed-on: https://chromium-review.googlesource.com/c/angle/angle/+/4628673
  Commit-Queue: Geoff Lang <geofflang@chromium.org>
  Reviewed-by: Shahbaz Youssefi <syoussefi@chromium.org>
  Reviewed-by: Quyen Le <lehoangquyen@chromium.org>
  

Download

• src/libANGLE/renderer/metal/mtl_pipeline_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_pipeline_cache.h:
  //    Defines classes for caching of mtl pipelines
  //
  
  #ifndef LIBANGLE_RENDERER_METAL_MTL_PIPELINE_CACHE_H_
  #define LIBANGLE_RENDERER_METAL_MTL_PIPELINE_CACHE_H_
  
  #include "common/hash_utils.h"
  #include "libANGLE/SizedMRUCache.h"
  #include "libANGLE/renderer/metal/mtl_utils.h"
  
  namespace rx
  {
  namespace mtl
  {
  
  struct PipelineKey
  {
      AutoObjCPtr<id<MTLFunction>> vertexShader;
      AutoObjCPtr<id<MTLFunction>> fragmentShader;
      RenderPipelineDesc pipelineDesc;
  
      AutoObjCPtr<id<MTLFunction>> computeShader;
  
      bool isRenderPipeline() const;
  
      bool operator==(const PipelineKey &rhs) const;
      size_t hash() const;
  };
  
  }  // namespace mtl
  }  // namespace rx
  
  namespace std
  {
  
  template <>
  struct hash<rx::mtl::PipelineKey>
  {
      size_t operator()(const rx::mtl::PipelineKey &key) const { return key.hash(); }
  };
  
  }  // namespace std
  
  namespace rx
  {
  namespace mtl
  {
  
  class PipelineCache : angle::NonCopyable
  {
    public:
      PipelineCache();
  
      angle::Result getRenderPipeline(ContextMtl *context,
                                      id<MTLFunction> vertexShader,
                                      id<MTLFunction> fragmentShader,
                                      const RenderPipelineDesc &desc,
                                      AutoObjCPtr<id<MTLRenderPipelineState>> *outRenderPipeline);
      angle::Result getComputePipeline(ContextMtl *context,
                                       id<MTLFunction> computeShader,
                                       AutoObjCPtr<id<MTLComputePipelineState>> *outComputePipeline);
  
    private:
      static constexpr unsigned int kMaxPipelines = 128;
  
      // The cache tries to clean up this many states at once.
      static constexpr unsigned int kGCLimit = 32;
  
      struct PipelineVariant
      {
          AutoObjCPtr<id<MTLRenderPipelineState>> renderPipeline;
          AutoObjCPtr<id<MTLComputePipelineState>> computePipeline;
      };
  
      using RenderPipelineMap = angle::base::HashingMRUCache<PipelineKey, PipelineVariant>;
      RenderPipelineMap mPipelineCache;
  };
  
  }  // namespace mtl
  }  // namespace rx
  
  #endif  // LIBANGLE_RENDERER_METAL_MTL_PIPELINE_CACHE_H_
  

Download