Add OS checks

Author: DenisVieriu97

backends/apple/mps/runtime/MPSDevice.h

@@ -32,6 +32,7 @@ enum class MacOSVersion : uint32_t {
   MACOS_VER_13_2_PLUS,
   MACOS_VER_13_3_PLUS,
   MACOS_VER_14_0_PLUS,
+  MACOS_VER_15_0_PLUS,
 };
 
 enum class LibraryType : uint32_t {
@@ -82,7 +83,8 @@ class MPSDevice {
   MPSDevice();
 };
 
-bool isMacOS13OrNewer(MacOSVersion version = MacOSVersion::MACOS_VER_13_0_PLUS);
+bool is_macos_13_or_newer(
+    MacOSVersion version = MacOSVersion::MACOS_VER_13_0_PLUS);
 
 } // namespace delegate
 } // namespace mps

backends/apple/mps/runtime/MPSDevice.mm

@@ -76,7 +76,7 @@ static inline MTLLanguageVersion getMetalLanguageVersion(const id<MTLDevice>& de
   static bool _macos_13_3_plus = [compileOptions respondsToSelector:@selector(maxTotalThreadsPerThreadgroup)] == YES;
 
   static bool _macos_14_0_plus = [mpsCD instancesRespondToSelector:@selector(conjugateWithTensor:name:)] == YES;
-
+  static bool _macos_15_0_plus = [mpsCD instancesRespondToSelector:@selector(scaledDotProductAttentionWithQueryTensor:keyTensor:valueTensor:maskTensor:scale:name:)] == YES;
   switch (version) {
     case MacOSVersion::MACOS_VER_13_0_PLUS:
       return _macos_13_0_plus;
@@ -88,6 +88,8 @@ static inline MTLLanguageVersion getMetalLanguageVersion(const id<MTLDevice>& de
       return _macos_13_3_plus;
     case MacOSVersion::MACOS_VER_14_0_PLUS:
       return _macos_14_0_plus;
+    case MacOSVersion::MACOS_VER_15_0_PLUS:
+      return _macos_15_0_plus;
     default:
       return false;
   }
@@ -144,7 +146,7 @@ static inline MTLLanguageVersion getMetalLanguageVersion(const id<MTLDevice>& de
   return err;
 }
 
-bool isMacOS13OrNewer(MacOSVersion version) {
+bool is_macos_13_or_newer(MacOSVersion version) {
   return MPSDevice::getInstance()->isMacOS13Plus(version);
 }
 

backends/apple/mps/runtime/MPSExecutor.mm

@@ -29,7 +29,7 @@ @interface MPSGraphExecutable()
 #if TARGET_OS_SIMULATOR or defined(__x86_64__)
   _use_shared_mem = false;
 #endif
-  if (!isMacOS13OrNewer(MacOSVersion::MACOS_VER_14_0_PLUS)) {
+  if (!is_macos_13_or_newer(MacOSVersion::MACOS_VER_14_0_PLUS)) {
     _use_shared_mem = false;
   }
 

backends/apple/mps/runtime/operations/BinaryOps.mm

@@ -119,7 +119,7 @@
     graphNode->output_id()                                                         \
   );                                                                               \
   ET_CHECK_OR_RETURN_ERROR(                                                        \
-    isMacOS13OrNewer(), NotSupported,                                              \
+    is_macos_13_or_newer(), NotSupported,                                              \
     "%s supported by MPS on MacOS13.0+/iOS16.1+", #aot_name);                       \
                                                                                    \
   _idToMPSGraphTensor[graphNode->output_id()] = binaryOpTensor(                    \
@@ -176,7 +176,7 @@
     return inputTensor;
   }
 
-  if (!isMacOS13OrNewer(MacOSVersion::MACOS_VER_13_0_PLUS)) {
+  if (!is_macos_13_or_newer(MacOSVersion::MACOS_VER_13_0_PLUS)) {
     MPSGraphTensor* zeroTensor = [mpsGraph constantWithScalar:0.0 dataType:inputTensor.dataType];
     MPSGraphTensor* predicateTensor = [mpsGraph lessThanWithPrimaryTensor:inputTensor
                                                           secondaryTensor:zeroTensor

backends/apple/mps/runtime/operations/QuantDequant.mm

@@ -22,19 +22,21 @@
     graphNode->output_id()
   );
 
+  ET_CHECK_OR_RETURN_ERROR(
+    is_macos_13_or_newer(MacOSVersion::MACOS_VER_15_0_PLUS),
+    NotImplemented,
+    "[ERROR] Operation %s is supported starting with macOS 15.0+ | iOS 18.0 + | iPadOS 18+ | tvOS 18+ | visionOS 2.0+ !",
+    mpsgraph::EnumNameMPSNodeUnion(nodePtr->mpsnode_union_type()));
+
   MPSGraphTensor* inputTensor = getMPSGraphTensor(graphNode->input1_id());
   MPSGraphTensor* scalesTensor = getMPSGraphTensor(graphNode->scales_id());
 
   MPSGraphTensor *zpTensor = [_mpsGraph constantWithScalar:0
                                                   dataType:MPSDataTypeInt4];
 
-  MPSGraphTensor *minTensor = [_mpsGraph constantWithScalar:0
-                                                  dataType:MPSDataTypeFloat16];
-
   MPSGraphTensor *wDqTensor = [_mpsGraph dequantizeTensor:inputTensor
                                               scaleTensor:scalesTensor
                                           zeroPointTensor:zpTensor
-                                                minTensor:minTensor
                                                 dataType:MPSDataTypeFloat16
                                                     name:nil];
 

backends/apple/mps/runtime/operations/UnaryOps.mm

@@ -33,7 +33,7 @@
     _idToMPSGraphTensor[graphNode->output_id()] = [_mpsGraph notWithTensor:inputTensor name:nil];
   } else {
     ET_CHECK_OR_RETURN_ERROR(
-      isMacOS13OrNewer(), NotSupported,
+      is_macos_13_or_newer(), NotSupported,
       "mpsBitwiseNotOp supported by MPS on MacOS13.0+/iOS16.1+");
     _idToMPSGraphTensor[graphNode->output_id()] = [_mpsGraph bitwiseNOTWithTensor:inputTensor name:nil];
   }