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 991481e commit 3f849dbCopy full SHA for 3f849db
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