We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1fef591 commit 81e95bbCopy full SHA for 81e95bb
test/inductor/test_cpp_wrapper_hipify.py
@@ -104,7 +104,11 @@ def test_hipify_aoti_driver_header(self) -> None:
104
}
105
"""
106
if torch.version.hip is not None:
107
- expected = expected.replace("32*numWarps", "64*numWarps")
+ # Adjusting the warp size to GPU supported wavefront size on AMD GPU
108
+ prop = torch.cuda.get_device_properties(torch.cuda.current_device())
109
+ expected = expected.replace(
110
+ "32*numWarps", str(prop.warp_size) + "*numWarps"
111
+ )
112
result = maybe_hipify_code_wrapper(header, True)
113
self.assertEqual(result.rstrip(), expected.rstrip())
114
0 commit comments