File tree Expand file tree Collapse file tree 2 files changed +11
-10
lines changed Expand file tree Collapse file tree 2 files changed +11
-10
lines changed Original file line number Diff line number Diff line change @@ -315,6 +315,7 @@ if(USE_CUDA)
315
315
list (APPEND JIT_TEST_SRCS ${NVFUSER_ROOT} /test /test_gpu_indexing_ops.cpp)
316
316
list (APPEND JIT_TEST_SRCS ${NVFUSER_ROOT} /test /test_gpu_indexing.cpp)
317
317
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)
318
319
319
320
add_executable (${NVFUSER_TESTS}
320
321
${TORCH_ROOT} /test /cpp/common/main.cpp
Original file line number Diff line number Diff line change 6
6
#include < gmock/gmock-matchers.h>
7
7
#include < gtest/gtest.h>
8
8
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>
17
17
18
18
#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>
21
21
22
22
#include < torch/torch.h>
23
23
You can’t perform that action at this time.
0 commit comments