Skip to content

Commit 0157db5

Browse files
committed
Remove unused enable_mkldnn for configure_system
1 parent d75da6c commit 0157db5

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

aarch64_linux/build_aarch64_wheel.py

+3-6
Original file line numberDiff line numberDiff line change
@@ -455,8 +455,7 @@ def build_torchaudio(host: RemoteHost, *,
455455
def configure_system(host: RemoteHost, *,
456456
compiler: str = "gcc-8",
457457
use_conda: bool = True,
458-
python_version: str = "3.8",
459-
enable_mkldnn: bool = False) -> None:
458+
python_version: str = "3.8") -> None:
460459
if use_conda:
461460
install_condaforge_python(host, python_version)
462461

@@ -514,8 +513,7 @@ def start_build(host: RemoteHost, *,
514513
configure_system(host,
515514
compiler=compiler,
516515
use_conda=use_conda,
517-
python_version=python_version,
518-
enable_mkldnn=enable_mkldnn)
516+
python_version=python_version)
519517
build_OpenBLAS(host, git_clone_flags)
520518
# build_FFTW(host, git_clone_flags)
521519

@@ -772,8 +770,7 @@ def parse_arguments():
772770
if args.use_torch_from_pypi:
773771
configure_system(host,
774772
compiler=args.compiler,
775-
python_version=python_version,
776-
enable_mkldnn=not args.disable_mkldnn)
773+
python_version=python_version)
777774
print("Installing PyTorch wheel")
778775
host.run_cmd("pip3 install torch")
779776
build_domains(host,

0 commit comments

Comments
 (0)