|
@@ -14,8 +14,9 @@ services:
|
|
|
environment:
|
|
environment:
|
|
|
COMPILE: ${COMPILE:-0}
|
|
COMPILE: ${COMPILE:-0}
|
|
|
PYTORCH_CUDA_ALLOC_CONF: ${PYTORCH_CUDA_ALLOC_CONF:-expandable_segments:True}
|
|
PYTORCH_CUDA_ALLOC_CONF: ${PYTORCH_CUDA_ALLOC_CONF:-expandable_segments:True}
|
|
|
- TORCH_LOGS: +dynamo
|
|
|
|
|
|
|
+ TORCH_LOGS: +dynamo,inductor
|
|
|
TORCHDYNAMO_VERBOSE: 1
|
|
TORCHDYNAMO_VERBOSE: 1
|
|
|
|
|
+ TORCH_COMPILE_DEBUG: 1
|
|
|
# GPU (remove this block if CPU-only):
|
|
# GPU (remove this block if CPU-only):
|
|
|
deploy:
|
|
deploy:
|
|
|
resources:
|
|
resources:
|
