We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f59b462 commit d2d888dCopy full SHA for d2d888d
torch/__init__.py
@@ -142,6 +142,10 @@ def _running_with_deploy() -> builtins.bool:
142
# Load the extension module
143
################################################################################
144
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
+
149
# If PyTorch was built against the ROCm runtime wheels, then there will be
150
# a _rocm_init module and it will define an initialize() function which can
151
# prepare ROCm for use. See general documentation on ROCm runtime wheels:
0 commit comments