Skip to content

Commit dd7d231

Browse files
henrylhtsangpytorchmergebot
authored andcommitted
[cutlass backend][test] re-enable test_cuda_compile_command for fbcode (pytorch#153001)
Differential Revision: [D74284047](https://our.internmc.facebook.com/intern/diff/D74284047/) Pull Request resolved: pytorch#153001 Approved by: https://github.com/ColinPeppler
1 parent 62b7ef0 commit dd7d231

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

test/inductor/test_codecache.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2139,7 +2139,6 @@ def test_get_hash_for_files(self):
21392139

21402140
class TestCudaCompileCommand(TestCase):
21412141
@unittest.skipIf(not HAS_CUDA, "Requires CUDA")
2142-
@unittest.skipIf(config.is_fbcode(), "fbcode requires different CUTLASS path setup")
21432142
def test_cuda_compile_command(self):
21442143
cmd_no_extra_args: str = cuda_compile_command(
21452144
["abc.cu", "def.cu"], "output", "so"
@@ -2161,7 +2160,7 @@ def test_cuda_compile_command(self):
21612160
CUDACodeCache.compile("test123.cu", "so", ["-Wsomething"])
21622161
check_output_mock.assert_called()
21632162
cmd_parts: list[str] = check_output_mock.call_args[0][0]
2164-
assert cmd_parts[0] == "nvcc", cmd_parts
2163+
assert cmd_parts[0].endswith("nvcc"), cmd_parts
21652164
assert "-Wsomething" in cmd_parts, cmd_parts
21662165
assert "-DNDEBUG" in cmd_parts, cmd_parts
21672166

0 commit comments

Comments
 (0)