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.
onednn-xpu
1 parent 7fc1507 commit 5dad575Copy full SHA for 5dad575
1 file changed
lib/torch-extension/default.nix
@@ -132,6 +132,7 @@ stdenv.mkDerivation (prevAttrs: {
132
++ lib.optionals rocmSupport (with rocmPackages; [ hipsparselt ])
133
++ lib.optionals xpuSupport ([
134
oneapi-torch-dev
135
+ xpuPackages.onednn-xpu
136
])
137
++ lib.optionals stdenv.hostPlatform.isDarwin [
138
apple-sdk_15
0 commit comments