From be71f9e202355269a74652e8f5f4bf3213c953d1 Mon Sep 17 00:00:00 2001
From: Anthony Shoumikhin <shoumikhin@meta.com>
Date: Thu, 22 May 2025 15:26:57 -0700
Subject: [PATCH] Fix diff-train by removing Float16 conformance

Summary: .

Reviewed By: Gasoonjia

Differential Revision: D75251897
---
 .../Exported/ExecuTorch+Tensor.swift          |  3 ---
 .../ExecuTorch/__tests__/TensorTest.swift     | 24 -------------------
 2 files changed, 27 deletions(-)

diff --git a/extension/apple/ExecuTorch/Exported/ExecuTorch+Tensor.swift b/extension/apple/ExecuTorch/Exported/ExecuTorch+Tensor.swift
index f1da13e7ae1..dc9ca543649 100644
--- a/extension/apple/ExecuTorch/Exported/ExecuTorch+Tensor.swift
+++ b/extension/apple/ExecuTorch/Exported/ExecuTorch+Tensor.swift
@@ -28,9 +28,6 @@ extension Int32: Scalar { static var dataType: DataType { .int } }
 extension Int64: Scalar { static var dataType: DataType { .long } }
 @available(*, deprecated, message: "This API is experimental.")
 extension Int: Scalar { static var dataType: DataType { .long } }
-@available(macOS 11.0, *)
-@available(*, deprecated, message: "This API is experimental.")
-extension Float16: Scalar { static var dataType: DataType { .half } }
 @available(*, deprecated, message: "This API is experimental.")
 extension Float: Scalar { static var dataType: DataType { .float } }
 @available(*, deprecated, message: "This API is experimental.")
diff --git a/extension/apple/ExecuTorch/__tests__/TensorTest.swift b/extension/apple/ExecuTorch/__tests__/TensorTest.swift
index 01fe094ea7a..4013cb2b296 100644
--- a/extension/apple/ExecuTorch/__tests__/TensorTest.swift
+++ b/extension/apple/ExecuTorch/__tests__/TensorTest.swift
@@ -172,30 +172,6 @@ class TensorTest: XCTestCase {
     }
   }
 
-  func testWithUnsafeBytesFloat16() throws {
-    var data: [Float16] = [1, 2, 3, 4, 5, 6]
-    let tensor = data.withUnsafeMutableBytes {
-      Tensor(bytesNoCopy: $0.baseAddress!, shape: [6], dataType: .half)
-    }
-    let array: [Float16] = try tensor.withUnsafeBytes { Array($0) }
-    XCTAssertEqual(array, data)
-  }
-
-  func testWithUnsafeMutableBytesFloat16() throws {
-    var data: [Float16] = [1, 2, 3, 4]
-    let tensor = data.withUnsafeMutableBytes { buffer in
-      Tensor(bytes: buffer.baseAddress!, shape: [4], dataType: .half)
-    }
-    try tensor.withUnsafeMutableBytes { (buffer: UnsafeMutableBufferPointer<Float16>) in
-      for i in buffer.indices {
-        buffer[i] *= 2
-      }
-    }
-    try tensor.withUnsafeBytes { buffer in
-      XCTAssertEqual(Array(buffer), data.map { $0 * 2 })
-    }
-  }
-
   func testInitWithTensor() {
     var data: [Int] = [10, 20, 30, 40]
     let tensor1 = data.withUnsafeMutableBytes {