chore: import upstream snapshot with attribution

2026-07-13 12:47:05 +08:00
commit 4f3b7da785
7394 changed files with 2005594 additions and 0 deletions
@@ -0,0 +1,71 @@
+/*
+ *  ******************************************************************************
+ *  *
+ *  *
+ *  * This program and the accompanying materials are made available under the
+ *  * terms of the Apache License, Version 2.0 which is available at
+ *  * https://www.apache.org/licenses/LICENSE-2.0.
+ *  *
+ *  * See the NOTICE file distributed with this work for additional
+ *  * information regarding copyright ownership.
+ *  * Unless required by applicable law or agreed to in writing, software
+ *  * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ *  * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+ *  * License for the specific language governing permissions and limitations
+ *  * under the License.
+ *  *
+ *  * SPDX-License-Identifier: Apache-2.0
+ *  *****************************************************************************
+ */
+
+//
+// @author raver119@gmail.com
+//
+#include <array/ConstantOffsetsBuffer.h>
+
+namespace sd {
+ConstantOffsetsBuffer::ConstantOffsetsBuffer(const std::shared_ptr<PointerWrapper> &primary)
+    : ConstantOffsetsBuffer(primary, std::shared_ptr<PointerWrapper>(nullptr)) {
+  //
+}
+
+ConstantOffsetsBuffer::ConstantOffsetsBuffer(const std::shared_ptr<PointerWrapper> &primary,
+                                             const std::shared_ptr<PointerWrapper> &special) {
+  _magic = MAGIC_VALID;  // Mark as valid/constructed
+  _primaryOffsets = primary;
+  _specialOffsets = special;
+}
+
+ConstantOffsetsBuffer::ConstantOffsetsBuffer() {
+  _magic = MAGIC_VALID;  // Mark default-constructed objects as valid
+  // shared_ptr members are automatically initialized to nullptr
+}
+
+ConstantOffsetsBuffer::~ConstantOffsetsBuffer() {
+  _magic = 0xDEADBEEF;  // Mark as destroyed - helps detect use-after-free
+}
+
+LongType *ConstantOffsetsBuffer::primary() {
+  // Check magic number first to detect use-after-free (dangling pointers)
+  if (!isValid()) {
+    THROW_EXCEPTION("ConstantOffsetsBuffer::primary: Object has been destroyed! Use-after-free detected (magic number check failed).");
+  }
+  if (_primaryOffsets == nullptr || !_primaryOffsets) {
+    THROW_EXCEPTION("ConstantOffsetsBuffer::primary: _primaryOffsets is nullptr! Buffer was likely already destroyed.");
+  }
+  return reinterpret_cast<LongType *>(_primaryOffsets->pointer());
+}
+
+LongType *ConstantOffsetsBuffer::special() {
+  return _specialOffsets ? reinterpret_cast<LongType *>(_specialOffsets->pointer()) : nullptr;
+}
+
+LongType *ConstantOffsetsBuffer::platform() {
+#ifdef SD_CUDA
+  return special();
+#else
+  return primary();
+#endif  // CUDABLAS
+}
+
+}  // namespace sd