liangsu9988's picture
Add package source and torch-universal build variant
b22e03e verified
Raw
History Blame Contribute Delete
142 Bytes
// SPDX-License-Identifier: Apache-2.0
#pragma once
#include <torch/all.h>
torch::Tensor flashrt_training_package_marker(torch::Tensor x);