oneapi-src / oneDNN

oneAPI Deep Neural Network Library (oneDNN)
https://uxlfoundation.org
Apache License 2.0
3.64k stars 1.01k forks source link

generic: conv: deconv: reduce kernel argument size #2182

Closed sgeor255 closed 2 weeks ago

sgeor255 commented 1 month ago

Description

This PR splits the conf struct for Convolution primitives into 3 separate structs, reducing the argument size passed to the kernel and enabling them to run on intel devices.

Checklist

General

mgouicem commented 1 month ago

make test disable device_cpu enable device_gpu enable thr_cuda enable thr_generic enable arch_rtx

sgeor255 commented 2 weeks ago

From what I can see the CI failure is not related to the changes in this PR.