Home / Class/ DynamicMetalShaderLibrary Class — pytorch Architecture

DynamicMetalShaderLibrary Class — pytorch Architecture

Architecture documentation for the DynamicMetalShaderLibrary class in MetalShaderLibrary.h from the pytorch codebase.

Entity Profile

Source Code

aten/src/ATen/native/mps/MetalShaderLibrary.h lines 189–196

class DynamicMetalShaderLibrary : public MetalShaderLibrary {
 public:
  DynamicMetalShaderLibrary(const std::string& src) : MetalShaderLibrary(src) {
    // Compile right away
    getLibrary();
  }
  ~DynamicMetalShaderLibrary() override;
};

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free