Skip to content
This repository was archived by the owner on Apr 28, 2023. It is now read-only.

Commit 9d4df32

Browse files
committed
Fix header includes
1 parent 09b2007 commit 9d4df32

File tree

4 files changed

+6
-4
lines changed

4 files changed

+6
-4
lines changed

tc/autotuner/genetic_autotuner.h

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,9 @@
1919
#include <memory>
2020
#include <vector>
2121

22-
#include "tc/autotuner/parameters.h"
22+
#include "tc/autotuner/genetic_tuning_harness.h"
23+
#include "tc/autotuner/utils/utils.h"
24+
#include "tc/core/execution_engine.h"
2325
#include "tc/lang/parser.h"
2426

2527
#include <llvm/ADT/Optional.h>

tc/autotuner/genetic_tuning_harness.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,10 +22,10 @@
2222
#include <unordered_map>
2323
#include <vector>
2424

25-
#include "tc/aten/aten_compiler.h"
2625
#include "tc/autotuner/genetic_search.h"
2726
#include "tc/autotuner/parameters.h"
2827
#include "tc/autotuner/utils/printer.h"
28+
#include "tc/core/utils/dlpack.h"
2929
#include "tc/lang/parser.h"
3030

3131
namespace tc {

tc/autotuner/utils/utils.h

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,6 @@
1616
#pragma once
1717
#include <vector>
1818

19-
#include <ATen/ATen.h>
20-
2119
#include "tc/core/cuda/cuda.h"
2220
#include "tc/core/cuda/cuda_mapping_options.h"
2321
#include "tc/core/utils/dlpack.h"

test/cuda/test_autotuner_utility.cc

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,8 @@
1515
*/
1616
#include <gtest/gtest.h>
1717

18+
#include "tc/aten/aten_compiler.h"
19+
#include "tc/aten/utils.h"
1820
#include "tc/autotuner/genetic_autotuner.h"
1921
#include "tc/autotuner/utils/utils.h"
2022
#include "tc/core/cuda/cuda_compilation_cache.h"

0 commit comments

Comments
 (0)