SPB Git

spb/forge Public MIT

Forge — LLM training from scratch in pure C++20 + Metal on Apple Silicon.

C++ 61.2% C 23% Python 7.6% TeX 7.2% CMake 1.1%
6.0 KB · 159 lines cpp
Raw Blame History
1//-------------------------------------------------------------------------------------------------------------------------------------------------------------2//3// Metal/MTLPrivate.hpp4//5// Copyright 2020-2024 Apple Inc.6//7// Licensed under the Apache License, Version 2.0 (the "License");8// you may not use this file except in compliance with the License.9// You may obtain a copy of the License at10//11//     http://www.apache.org/licenses/LICENSE-2.012//13// Unless required by applicable law or agreed to in writing, software14// distributed under the License is distributed on an "AS IS" BASIS,15// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.16// See the License for the specific language governing permissions and17// limitations under the License.18//19//-------------------------------------------------------------------------------------------------------------------------------------------------------------2021#pragma once2223//-------------------------------------------------------------------------------------------------------------------------------------------------------------2425#include "MTLDefines.hpp"2627#include <objc/runtime.h>2829//-------------------------------------------------------------------------------------------------------------------------------------------------------------3031#define _MTL_PRIVATE_CLS(symbol) (MTL::Private::Class::s_k##symbol)32#define _MTL_PRIVATE_SEL(accessor) (MTL::Private::Selector::s_k##accessor)3334//-------------------------------------------------------------------------------------------------------------------------------------------------------------3536#if defined(MTL_PRIVATE_IMPLEMENTATION)3738#ifdef METALCPP_SYMBOL_VISIBILITY_HIDDEN39#define _MTL_PRIVATE_VISIBILITY __attribute__((visibility("hidden")))40#else41#define _MTL_PRIVATE_VISIBILITY __attribute__((visibility("default")))42#endif // METALCPP_SYMBOL_VISIBILITY_HIDDEN4344#define _MTL_PRIVATE_IMPORT __attribute__((weak_import))4546#ifdef __OBJC__47#define _MTL_PRIVATE_OBJC_LOOKUP_CLASS(symbol) ((__bridge void*)objc_lookUpClass(#symbol))48#define _MTL_PRIVATE_OBJC_GET_PROTOCOL(symbol) ((__bridge void*)objc_getProtocol(#symbol))49#else50#define _MTL_PRIVATE_OBJC_LOOKUP_CLASS(symbol) objc_lookUpClass(#symbol)51#define _MTL_PRIVATE_OBJC_GET_PROTOCOL(symbol) objc_getProtocol(#symbol)52#endif // __OBJC__5354#define _MTL_PRIVATE_DEF_CLS(symbol) void* s_k##symbol _MTL_PRIVATE_VISIBILITY = _MTL_PRIVATE_OBJC_LOOKUP_CLASS(symbol)55#define _MTL_PRIVATE_DEF_PRO(symbol) void* s_k##symbol _MTL_PRIVATE_VISIBILITY = _MTL_PRIVATE_OBJC_GET_PROTOCOL(symbol)56#define _MTL_PRIVATE_DEF_SEL(accessor, symbol) SEL s_k##accessor _MTL_PRIVATE_VISIBILITY = sel_registerName(symbol)5758#include <dlfcn.h>59#define MTL_DEF_FUNC( name, signature ) \60    using Fn##name = signature; \61    Fn##name name = reinterpret_cast< Fn##name >( dlsym( RTLD_DEFAULT, #name ) )6263namespace MTL::Private64{65    template <typename _Type>66    inline _Type const LoadSymbol(const char* pSymbol)67    {68        const _Type* pAddress = static_cast<_Type*>(dlsym(RTLD_DEFAULT, pSymbol));6970        return pAddress ? *pAddress : nullptr;71    }72} // MTL::Private7374#if (TARGET_OS_OSX    && __MAC_26_0   && (__MAC_OS_X_VERSION_MIN_REQUIRED  >= __MAC_26_0))                           || \75    (TARGET_OS_IPHONE && _IPHONE_26_0 && (__IPHONE_OS_VERSION_MIN_REQUIRED >= __IPHONE_26_0) && (!TARGET_OS_VISION)) || \76    (TARGET_OS_TV     && __TVOS_26_0  && (__TV_OS_VERSION_MIN_REQUIRED     >= __TVOS_26_0))7778#define _MTL_PRIVATE_DEF_STR(type, symbol)                  \79    _MTL_EXTERN type const MTL##symbol _MTL_PRIVATE_IMPORT; \80    type const                         MTL::symbol = (nullptr != &MTL##symbol) ? MTL##symbol : nullptr8182#define _MTL_PRIVATE_DEF_CONST(type, symbol)              \83    _MTL_EXTERN type const MTL##symbol _MTL_PRIVATE_IMPORT; \84    type const                         MTL::symbol = (nullptr != &MTL##symbol) ? MTL##symbol : nullptr8586#define _MTL_PRIVATE_DEF_WEAK_CONST(type, symbol) \87    _MTL_EXTERN type const MTL##symbol;    \88    type const             MTL::symbol = MTL::Private::LoadSymbol<type>("MTL" #symbol)8990#else9192#define _MTL_PRIVATE_DEF_STR(type, symbol) \93    _MTL_EXTERN type const MTL##symbol;    \94    type const             MTL::symbol = MTL::Private::LoadSymbol<type>("MTL" #symbol)9596#define _MTL_PRIVATE_DEF_CONST(type, symbol) \97    _MTL_EXTERN type const MTL##symbol;    \98    type const             MTL::symbol = MTL::Private::LoadSymbol<type>("MTL" #symbol)99100#define _MTL_PRIVATE_DEF_WEAK_CONST(type, symbol) _MTL_PRIVATE_DEF_CONST(type, symbol)101102#endif103104#else105106#define _MTL_PRIVATE_DEF_CLS(symbol) extern void* s_k##symbol107#define _MTL_PRIVATE_DEF_PRO(symbol) extern void* s_k##symbol108#define _MTL_PRIVATE_DEF_SEL(accessor, symbol) extern SEL s_k##accessor109#define _MTL_PRIVATE_DEF_STR(type, symbol) extern type const MTL::symbol110#define _MTL_PRIVATE_DEF_CONST(type, symbol) extern type const MTL::symbol111#define _MTL_PRIVATE_DEF_WEAK_CONST(type, symbol) extern type const MTL::symbol112113#endif // MTL_PRIVATE_IMPLEMENTATION114115//-------------------------------------------------------------------------------------------------------------------------------------------------------------116117namespace MTL118{119namespace Private120{121    namespace Class122    {123124    } // Class125} // Private126} // MTL127128//-------------------------------------------------------------------------------------------------------------------------------------------------------------129130namespace MTL131{132namespace Private133{134    namespace Protocol135    {136137    } // Protocol138} // Private139} // MTL140141//-------------------------------------------------------------------------------------------------------------------------------------------------------------142143namespace MTL144{145namespace Private146{147    namespace Selector148    {149150        _MTL_PRIVATE_DEF_SEL(beginScope,151            "beginScope");152        _MTL_PRIVATE_DEF_SEL(endScope,153            "endScope");154    } // Class155} // Private156} // MTL157158//-------------------------------------------------------------------------------------------------------------------------------------------------------------159