Skip to content

Commit

Permalink
enable conv_1x1_as_mm option for torch inductor in diffusers tests
Browse files Browse the repository at this point in the history
- enabling conv_1x1_as_mm option removes cuda illegal memory access error that occurs during the tiny stable diffusion model's compilation on CI machine or nvidia T4 gpu
  • Loading branch information
aliabdelkader committed Mar 21, 2024
1 parent 9baed53 commit 54a1c09
Showing 1 changed file with 5 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -9,3 +9,8 @@ defaults:
- _self_ # hydra 1.1 compatibility

experiment_name: cuda_inference_pytorch_diffusers_torch_compile

backend:
torch_compile_config:
options:
conv_1x1_as_mm: True

0 comments on commit 54a1c09

Please sign in to comment.