Add assert that is_parallel is true

ghstack-source-id: c8f168ba52
Pull Request resolved: https://github.com/pytorch/pytorch/pull/146779
This commit is contained in:
James Wu 2025-02-08 20:15:31 -08:00
parent b8c5c4443c
commit 7c537fdff1

View file

@ -289,6 +289,7 @@ class AsyncCompile:
_load_triton_kernel_from_source, kernel_name, source_code
)
is_parallel = self.use_process_pool()
assert(is_parallel)
set_feature_use("parallel_compile_post_warmup", is_parallel)
if is_parallel:
# We want to support changing these env vars after (and while) the