Skip to content

Conversation

@Saransh-cpp
Copy link

@Saransh-cpp Saransh-cpp commented Dec 9, 2025

Adds transpose kernel for the alpaka backend with the following signature:

template <typename TAcc, typename T>
ALPAKA_FN_ACC void operator()(TAcc const& acc, T const* input, T* output, const std::size_t* input_strides,
                              const std::size_t* output_strides, const std::size_t* input_shape,
                              const std::size_t* output_shape, const std::size_t* perm,
                              const std::size_t ndim) const {

I am not sure if there is a concrete way to test this, but I have added tests for the kernel itself in a stand-alone repository - https://github.com/Saransh-cpp/SOFIE-ALPAKA/blob/main/kernels/transpose.hpp + https://github.com/Saransh-cpp/SOFIE-ALPAKA/blob/main/tests/test_transpose.cpp.

cc: @sanjibansg

@Saransh-cpp Saransh-cpp force-pushed the feature/alpaka-transpose branch from 54d227d to f45aa79 Compare December 9, 2025 13:21
@Saransh-cpp Saransh-cpp force-pushed the feature/alpaka-transpose branch from daddbbf to a4677ec Compare December 15, 2025 22:17
@Saransh-cpp
Copy link
Author

Errors with:

[ERROR] Failed processing /mnt/d/Saransh/HEP/SOFIE/src/SOFIE_core/test/input_models/Transpose.onnx: TMVA SOFIE tensor [input] is an input tensor with unspecified dimension parameter

locally when run with the new testing suite.

@Saransh-cpp
Copy link
Author

Ahh, this is coming from the Transpose.onnx model and not the kernel.

@Saransh-cpp
Copy link
Author

Also probably needs adding a reference output for the tests

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant