Skip to content

Commit

Permalink
Import MKL via //third-party/mkl targets (#126371)
Browse files Browse the repository at this point in the history
Summary:
This is a step towards upgrading the MKL library and using a buckified targets rather than importing from TP2.

- Add new `//third-party/mkl:mkl_xxx` targets that are currently aliases to `third-party//IntelComposerXE:mkl_xxx`.
- Switch usage of `external_deps = [("IntelComposerXE", None, "mkl_xxx")]` to `deps = ["fbsource//third-party/mkl:mkl_xxx"]`

Note that this only changes references to `mkl_xxx` references in `IntelComposerXE` but not references to "svml" or "ipp*".

Test Plan: sandcastle

Differential Revision: D57360438

Pull Request resolved: #126371
Approved by: https://github.com/bertmaher
  • Loading branch information
MatzeB authored and pytorchmergebot committed May 16, 2024
1 parent 62403b5 commit 796dff7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion defs.bzl
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
def get_blas_gomp_arch_deps():
return [
("x86_64", [
"third-party//IntelComposerXE:{}".format(native.read_config("fbcode", "mkl_lp64", "mkl_lp64_omp")),
"fbsource//third-party/mkl:{}".format(native.read_config("fbcode", "mkl_lp64", "mkl_lp64_omp")),
]),
("aarch64", [
"third-party//OpenBLAS:OpenBLAS",
Expand Down

0 comments on commit 796dff7

Please sign in to comment.