We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent dcf2a55 commit 27d6fc6Copy full SHA for 27d6fc6
torchao/csrc/cpu/int8_sdpa.cpp
@@ -1,14 +1,14 @@
1
-#pragma once
2
#include <ATen/ATen.h>
3
#include <ATen/AccumulateType.h>
4
#include <ATen/Dispatch.h>
5
-#include <ATen/Parallel.h>
6
#include <ATen/core/Tensor.h>
7
#include <ATen/cpu/vec/functional.h>
8
#include <ATen/cpu/vec/vec.h>
9
#include <ATen/cpu/Utils.h>
10
-
11
#include <ATen/native/cpu/utils.h>
+#include <ATen/native/CPUBlas.h>
+#include <ATen/Parallel.h>
+#include <ATen/Tensor.h>
12
#include <c10/util/irange.h>
13
14
#ifndef AT_PER_OPERATOR_HEADERS
@@ -17,12 +17,10 @@
17
#include <ATen/ops/empty.h>
18
#endif
19
20
-#include <ATen/Tensor.h>
21
#include <limits>
22
#include <omp.h>
23
#include <torch/all.h>
24
#include <torch/csrc/autograd/function.h>
25
-#include <ATen/native/CPUBlas.h>
26
27
namespace torchao {
28
0 commit comments