composable_kernel
composable_kernel copied to clipboard
elementwise operation Support 1D~5D
We only support 2D (DeviceBinaryElementWise2D) so far.
Support more dimension for flexibility
https://ontrack-internal.amd.com/browse/LWPCK-190
more specifically, we should have a single device operation class that support 1D~5D tensor elementwise operation, instead of 5 device operation classes.
@rocking5566 Please advise if we can go ahead and close this ticket as fixed. Thanks!
@ppanchad-amd As mention in the https://ontrack-internal.amd.com/browse/LWPCK-190 We can close this ticket