torch-ext/torch_binding.h

// SPDX-License-Identifier: Apache-2.0

#pragma once

#include <torch/all.h>

torch::Tensor flashrt_training_package_marker(torch::Tensor x);