Skip to content

GitLab

"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "3956d90eb6b2d2ab280787fba887a4868a51f0e4"
Commit 52ae168b authored by Chao Liu's avatar Chao Liu
Browse files

adding inline asm

parent fdaaaa50
Hide whitespace changes
Inline Side-by-side
Showing with 13 additions and 1 deletion
driver/driver.hip.cpp
View file @ 52ae168b
...@@ -580,7 +580,7 @@ int main(int argc, char* argv[]) ...@@ -580,7 +580,7 @@ int main(int argc, char* argv[])
constexpr unsigned HPad = 0; constexpr unsigned HPad = 0;
constexpr unsigned WPad = 0; constexpr unsigned WPad = 0;
#elif 1 #elif 0
// 1x1 filter, 14x14 image // 1x1 filter, 14x14 image
constexpr unsigned N = 128; constexpr unsigned N = 128;
constexpr unsigned C = 2048; constexpr unsigned C = 2048;
...@@ -590,6 +590,18 @@ int main(int argc, char* argv[]) ...@@ -590,6 +590,18 @@ int main(int argc, char* argv[])
constexpr unsigned Y = 1; constexpr unsigned Y = 1;
constexpr unsigned X = 1; constexpr unsigned X = 1;
constexpr unsigned HPad = 0;
constexpr unsigned WPad = 0;
#elif 1
// 1x1 filter, 14x14 image, C = 256
constexpr unsigned N = 128;
constexpr unsigned C = 256;
constexpr unsigned HI = 14;
constexpr unsigned WI = 14;
constexpr unsigned K = 512;
constexpr unsigned Y = 1;
constexpr unsigned X = 1;
constexpr unsigned HPad = 0; constexpr unsigned HPad = 0;
constexpr unsigned WPad = 0; constexpr unsigned WPad = 0;
#endif #endif
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment