Skip to content

Commit a06224f

Browse files
committed
Modify CMakeLists.txt and headers following rebase.
This is just the necessary changes to get my tests compiling after Jie's big refactor, which I've just rebased in.
1 parent 79bb154 commit a06224f

File tree

2 files changed

+11
-10
lines changed

2 files changed

+11
-10
lines changed

third_party/nvfuser/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -315,6 +315,7 @@ if(USE_CUDA)
315315
list(APPEND JIT_TEST_SRCS ${NVFUSER_ROOT}/test/test_gpu_indexing_ops.cpp)
316316
list(APPEND JIT_TEST_SRCS ${NVFUSER_ROOT}/test/test_gpu_indexing.cpp)
317317
list(APPEND JIT_TEST_SRCS ${NVFUSER_ROOT}/test/test_gpu_gather_ops.cpp)
318+
list(APPEND JIT_TEST_SRCS ${NVFUSER_ROOT}/test/test_gpu_match_frontend.cpp)
318319

319320
add_executable(${NVFUSER_TESTS}
320321
${TORCH_ROOT}/test/cpp/common/main.cpp

third_party/nvfuser/test/test_gpu_match_frontend.cpp

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -6,18 +6,18 @@
66
#include <gmock/gmock-matchers.h>
77
#include <gtest/gtest.h>
88

9-
#include <torch/csrc/jit/codegen/cuda/arith.h>
10-
#include <torch/csrc/jit/codegen/cuda/executor.h>
11-
#include <torch/csrc/jit/codegen/cuda/inlining.h>
12-
#include <torch/csrc/jit/codegen/cuda/ir_all_nodes.h>
13-
#include <torch/csrc/jit/codegen/cuda/ir_builder.h>
14-
#include <torch/csrc/jit/codegen/cuda/kernel_cache.h>
15-
#include <torch/csrc/jit/codegen/cuda/scheduler/all_schedulers.h>
16-
#include <torch/csrc/jit/codegen/cuda/scheduler/utils.h>
9+
#include <arith.h>
10+
#include <executor.h>
11+
#include <inlining.h>
12+
#include <ir_all_nodes.h>
13+
#include <ir_builder.h>
14+
#include <kernel_cache.h>
15+
#include <scheduler/all_schedulers.h>
16+
#include <scheduler/utils.h>
1717

1818
#include <test/cpp/jit/test_utils.h>
19-
#include <torch/csrc/jit/codegen/cuda/test/test_gpu_validator.h>
20-
#include <torch/csrc/jit/codegen/cuda/test/test_utils.h>
19+
#include <test/test_gpu_validator.h>
20+
#include <test/test_utils.h>
2121

2222
#include <torch/torch.h>
2323

0 commit comments

Comments
 (0)