[CPU] remove RECORD_FUNCTION (#23528)

This commit is contained in:
Ma Mingfei
2026-04-24 09:18:29 +08:00
committed by GitHub
parent 87e50f20f6
commit 23e4d381f0
21 changed files with 0 additions and 106 deletions
-4
View File
@@ -48,8 +48,6 @@ void initialize(int64_t size, int64_t rank) {
}
void shm_allreduce(torch::Tensor& data, int64_t op) {
RECORD_FUNCTION("sgl-kernel::shm_allreduce", std::vector<c10::IValue>({data}));
TORCH_CHECK(op == c10d::ReduceOp::SUM, "Only torch.distributed.ReduceOp.SUM is supported");
auto numel = data.numel();
@@ -60,8 +58,6 @@ void shm_allreduce(torch::Tensor& data, int64_t op) {
}
torch::Tensor shm_allgather(torch::Tensor& data, int64_t dim) {
RECORD_FUNCTION("sgl-kernel::shm_allgather", std::vector<c10::IValue>({data}));
auto numel = data.numel();
int data_size = numel * data.element_size();
if (dim < 0) {