Skip to content

Commit 27d6fc6

Browse files
committed
fix internal build errors
1 parent dcf2a55 commit 27d6fc6

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

torchao/csrc/cpu/int8_sdpa.cpp

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,14 @@
1-
#pragma once
21
#include <ATen/ATen.h>
32
#include <ATen/AccumulateType.h>
43
#include <ATen/Dispatch.h>
5-
#include <ATen/Parallel.h>
64
#include <ATen/core/Tensor.h>
75
#include <ATen/cpu/vec/functional.h>
86
#include <ATen/cpu/vec/vec.h>
97
#include <ATen/cpu/Utils.h>
10-
118
#include <ATen/native/cpu/utils.h>
9+
#include <ATen/native/CPUBlas.h>
10+
#include <ATen/Parallel.h>
11+
#include <ATen/Tensor.h>
1212
#include <c10/util/irange.h>
1313

1414
#ifndef AT_PER_OPERATOR_HEADERS
@@ -17,12 +17,10 @@
1717
#include <ATen/ops/empty.h>
1818
#endif
1919

20-
#include <ATen/Tensor.h>
2120
#include <limits>
2221
#include <omp.h>
2322
#include <torch/all.h>
2423
#include <torch/csrc/autograd/function.h>
25-
#include <ATen/native/CPUBlas.h>
2624

2725
namespace torchao {
2826

0 commit comments

Comments
 (0)