Skip to content

[ML] Upgrade to PyTorch 1.11 on macOS#2235

Merged
droberts195 merged 1 commit intoelastic:mainfrom
droberts195:mac_torch_1.11
Mar 30, 2022
Merged

[ML] Upgrade to PyTorch 1.11 on macOS#2235
droberts195 merged 1 commit intoelastic:mainfrom
droberts195:mac_torch_1.11

Conversation

@droberts195
Copy link
Copy Markdown

It's now possible to use the MKLDNN library on macOS on ARM.

There are no other changes to the build process since 1.9.

I've added a comment to say we should upgrade the version of
Python we use to build PyTorch before the next upgrade, as
this seems to be the root cause of the difficulties with
correct architecture detection on ARM.

It's now possible to use the MKLDNN library on macOS on ARM.

There are no other changes to the build process since 1.9.

I've added a comment to say we should upgrade the version of
Python we use to build PyTorch before the next upgrade, as
this seems to be the root cause of the difficulties with
correct architecture detection on ARM.
Copy link
Copy Markdown
Contributor

@dimitris-athanasiou dimitris-athanasiou left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM Ah, just realized this is still a draft. I'll check it out again when it's ready for review.

droberts195 added a commit to droberts195/ml-cpp that referenced this pull request Mar 22, 2022
Together with elastic#2233 and elastic#2235 this completes the upgrade to
PyTorch 1.11 for all platforms.

A small change to the way things are done is that we now copy
all MKL libraries into the Docker images, then copy the required
subset into our release distribution. This means we can change
the subset of MKL libraries that we redistribute without having
to rebuild our Docker images.

Fixes elastic#2126
@droberts195 droberts195 marked this pull request as ready for review March 28, 2022 08:23
Copy link
Copy Markdown
Member

@davidkyle davidkyle left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@droberts195 droberts195 merged commit 64aa225 into elastic:main Mar 30, 2022
@droberts195 droberts195 deleted the mac_torch_1.11 branch March 30, 2022 07:48
droberts195 added a commit that referenced this pull request Mar 30, 2022
Together with #2233 and #2235 this completes the upgrade to
PyTorch 1.11 for all platforms.

A small change to the way things are done is that we now copy
all MKL libraries into the Docker images, then copy the required
subset into our release distribution. This means we can change
the subset of MKL libraries that we redistribute without having
to rebuild our Docker images.

Fixes #2126
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants