Skip to content

Commit d2d888d

Browse files
committed
Set MIOpen env variables to disable direct naive and some winograd fury conv kernels
1 parent f59b462 commit d2d888d

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

torch/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -142,6 +142,10 @@ def _running_with_deploy() -> builtins.bool:
142142
# Load the extension module
143143
################################################################################
144144

145+
# Set MIOpen environment variables
146+
os.environ.setdefault("MIOPEN_DEBUG_CONV_DIRECT_NAIVE_CONV_FWD", "0")
147+
os.environ.setdefault("MIOPEN_DEBUG_AMD_WINOGRAD_FURY_RXS_F2X3", "0")
148+
145149
# If PyTorch was built against the ROCm runtime wheels, then there will be
146150
# a _rocm_init module and it will define an initialize() function which can
147151
# prepare ROCm for use. See general documentation on ROCm runtime wheels:

0 commit comments

Comments
 (0)