Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

[WIP] Transpose 2D GPU #16706

Closed
wants to merge 12 commits into from
Closed

[WIP] Transpose 2D GPU #16706

wants to merge 12 commits into from

Conversation

ChaiBapchya
Copy link
Contributor

@ChaiBapchya ChaiBapchya commented Nov 1, 2019

Description

2D Transpose for GPU

Checklist

Essentials

Please feel free to remove inapplicable items for your PR.

  • Changes are complete (i.e. I finished coding on this PR)
  • All changes have test coverage:
  • Code is well-documented:
  • To the my best knowledge, examples are either not affected by this change, or have been fixed to be compatible with this change

Changes

  • modified: src/operator/tensor/matrix_op-inl.h
  • modified: src/operator/tensor/matrix_op.cu
  • modified: src/operator/tensor/matrix_op.cc

Comments

Replaces the mshadow expression template (used for GPU context) for 2D Transpose with this implementation
Follows-up on #16104

Only issue - Upon building on UNIX GPU - warning

/home/ubuntu/workspace/incubator-mxnet/src/operator/numpy/./../tensor/matrix_op-inl.h(266): warning: extern inline function "mxnet::op::Transpose2D<DType,xpu>(const DType *, DType *, mxnet::index_t, mxnet::index_t) [with DType=mshadow::half::ha
lf_t, xpu=mxnet::gpu]" was referenced but not defined

@ChaiBapchya
Copy link
Contributor Author

Closing since de-prioritized.

@ChaiBapchya ChaiBapchya deleted the 2dgpu branch September 18, 2020 17:11
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant