File tree Expand file tree Collapse file tree 3 files changed +0
-8
lines changed
torchrec/inference/inference_legacy/src Expand file tree Collapse file tree 3 files changed +0
-8
lines changed Original file line number Diff line number Diff line change 18
18
19
19
#include < ATen/Functions.h> // @manual
20
20
#include < ATen/core/Dict.h>
21
- #include < ATen/core/interned_strings.h>
22
21
#include < ATen/record_function.h> // @manual
23
22
#include < c10/core/Device.h>
24
23
#include < c10/core/DeviceType.h>
25
- #include < c10/cuda/CUDAFunctions.h>
26
24
#include < c10/cuda/CUDAGuard.h>
27
25
#include < c10/cuda/CUDAStream.h>
28
26
#include < fmt/format.h>
Original file line number Diff line number Diff line change 16
16
#include < c10/cuda/CUDAGuard.h>
17
17
#include < fmt/format.h>
18
18
#include < folly/MPMCQueue.h>
19
- #include < folly/ScopeGuard.h>
20
- #include < folly/Synchronized.h>
21
19
#include < folly/executors/CPUThreadPoolExecutor.h>
22
20
#include < folly/futures/Future.h>
23
- #include < folly/io/IOBuf.h>
24
21
#include < folly/io/async/Request.h>
25
22
#include < folly/stop_watch.h>
26
23
#include < gflags/gflags.h>
35
32
#endif
36
33
37
34
#include " ATen/cuda/CUDAEvent.h"
38
- #include " torchrec/inference/BatchingQueue.h"
39
35
#include " torchrec/inference/ExceptionHandler.h"
40
36
#include " torchrec/inference/Observer.h"
41
37
#include " torchrec/inference/Types.h"
Original file line number Diff line number Diff line change 8
8
9
9
#include " torchrec/inference/ResultSplit.h"
10
10
11
- #include < c10/core/ScalarType.h>
12
11
#include < folly/Range.h>
13
12
#include < folly/container/Enumerate.h>
14
13
#include < folly/io/Cursor.h>
15
14
16
- #include " ATen/Functions.h"
17
15
#include " torchrec/inference/Types.h"
18
16
19
17
namespace torchrec {
You can’t perform that action at this time.
0 commit comments