mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
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:
parent
b8c5c4443c
commit
7c537fdff1
1 changed files with 1 additions and 0 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in a new issue