mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
2ee4ef5290
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/13740 We would like to rename the old fbgemm to “fbgemm0”, and the new fbgemm2 to “fbgemm”: This DIFF changes all namespace fbgemm2 to namespace fbgemm. The purpose is to avoid the confusion of "fbgemm2" when we release our FBGEMM open source. Reviewed By: jspark1105 Differential Revision: D12850449 fbshipit-source-id: 08cc47864b157e36fbceddb7a10bf26218c67bd8
14 lines
285 B
C++
14 lines
285 B
C++
#pragma once
|
|
|
|
#include <cstdint>
|
|
|
|
namespace fbgemm {
|
|
|
|
/**
|
|
* Transpose 4xN matrix with unsigned 8-byte integers
|
|
* TODO: move this to fbgemm after making transpose routine more general
|
|
*/
|
|
void transpose_4rows(int N, const std::uint8_t* src, std::uint8_t* dst);
|
|
|
|
} // namespace fbgemm
|