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)
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
Original file line number Diff line number Diff line change 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
You can’t perform that action at this time.
0 commit comments