[WIP] Enable the transform tensor API to convert any NCHW tensor to NHWC tensor and vice versa #2003
+554
−288
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
atamazov
suggested changes
Mar 2, 2023
| return a.first < b.first; | ||
| }); | ||
|
|
||
| printf("Permutation for %s -> %s: ", src_layout.c_str(), dst_layout.c_str()); |
There was a problem hiding this comment.
Use MIOPEN_LOG_I2/I/W/E macros, no printf please.
…nd ROCm support in CMake
…nstall the extra gputt dependency. It should be built like this: BUILDKIT_PROGRESS=plain docker build --no-cache -f Dockerfile.gputt -t rocm/miopen:ci-gputt . Then the usual miopen:ci invocation should instead use miopen:ci-gputt: docker run -it -v $HOME:$HOME --privileged --rm --device=/dev/kfd --device /dev/dri:/dev/dri:rw --volume /dev/dri:/dev/dri:rw -v /var/lib/docker/:/var/lib/docker --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined rocm/miopen:ci-gputt fish
dmikushin
force-pushed
the
enable-hchw-nhwc-conversion
branch
from
f1d20f7 to
c686acf
Compare
Co-authored-by: Vasilii Filippov <[email protected]>
…or transform by gpuTT functionality. In order for that gpuTT has been extended to support all data types required by MIOpen, including the fancy int8x4
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
LWPMIOPEN-125
We substitute the TensorTransfom implementation with a more generic GPU-based gpuTT library calls.
Few things still to be implemented: