ZYM/BH-int2048-2023
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

upd: in the way of opt

Browse Source
This commit is contained in:
ZhuangYumin
2023-10-31 18:16:55 +08:00
parent db53bfdcff
commit f2ae81fb54
3 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
data/CMakeLists.txt
View File

@ -1,6 +1,6 @@
set(PROJECT_NAME ${CMAKE_PROJECT_NAME}) set(PROJECT_NAME ${CMAKE_PROJECT_NAME})
set(CMAKE_CXX_STANDARD 17) set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CXX_FLAGS "-g") set(CMAKE_CXX_FLAGS "-O2")
set(ENV{MAKEFLAGS} "-j16") set(ENV{MAKEFLAGS} "-j16")
include_directories(${PROJECT_SOURCE_DIR}/include) include_directories(${PROJECT_SOURCE_DIR}/include)
link_directories(${PROJECT_SOURCE_DIR}/src) link_directories(${PROJECT_SOURCE_DIR}/src)

2
src/CMakeLists.txt
View File

@ -1,5 +1,5 @@
set(PROJECT_NAME ${CMAKE_PROJECT_NAME}) set(PROJECT_NAME ${CMAKE_PROJECT_NAME})
set(CMAKE_CXX_STANDARD 17) set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CXX_FLAGS "-g") set(CMAKE_CXX_FLAGS "-O2")
include_directories(${PROJECT_SOURCE_DIR}/include) include_directories(${PROJECT_SOURCE_DIR}/include)
add_library(int2048 STATIC int2048.cpp) add_library(int2048 STATIC int2048.cpp)

10
src/int2048.cpp
View File

@ -506,8 +506,11 @@ inline void UnsignedMultiply(int2048 &A, const int2048 *pB,
int blocks_of_A = ((A.num_length + int2048::kNum - 1) / int2048::kNum); int blocks_of_A = ((A.num_length + int2048::kNum - 1) / int2048::kNum);
int blocks_of_B = ((pB->num_length + int2048::kNum - 1) / int2048::kNum); int blocks_of_B = ((pB->num_length + int2048::kNum - 1) / int2048::kNum);
if (inverse) { if (inverse) {
assert(pB->num_length % int2048::kNum == 0);
lenght_limit = std::min(lenght_limit, pB->num_length); lenght_limit = std::min(lenght_limit, pB->num_length);
blocks_of_B = lenght_limit; blocks_of_B = lenght_limit / int2048::kNum;
// assert(blocks_of_B ==
// ((pB->num_length + int2048::kNum - 1) / int2048::kNum));
} }
int max_blocks = blocks_of_A + blocks_of_B; int max_blocks = blocks_of_A + blocks_of_B;
int NTT_blocks = 2; int NTT_blocks = 2;
@ -708,7 +711,8 @@ inline void UnsignedDivide(int2048 &A, const int2048 *pB) {
} }
assert(tmp_x.num_length % int2048::kNum == 0); assert(tmp_x.num_length % int2048::kNum == 0);
assert(inverse_B.num_length % int2048::kNum == 0); assert(inverse_B.num_length % int2048::kNum == 0);
UnsignedMultiply(tmp_x, &inverse_B, true, inverse_B.num_length); UnsignedMultiply(tmp_x, &inverse_B, true,
tmp_x.num_length + 3 * int2048::kNum);
for (int i = 0; i < tmp_x_error + inverseB_error; i++) for (int i = 0; i < tmp_x_error + inverseB_error; i++)
tmp_x.RestoreHalfBlock(); tmp_x.RestoreHalfBlock();
UnsignedMinus(inverse_two, &tmp_x, true); UnsignedMinus(inverse_two, &tmp_x, true);
@ -721,7 +725,7 @@ inline void UnsignedDivide(int2048 &A, const int2048 *pB) {
x_error = 1; x_error = 1;
x.ProcessHalfBlock(); x.ProcessHalfBlock();
} }
UnsignedMultiply(x, &inverse_two, true); UnsignedMultiply(x, &inverse_two, true, inverse_two.num_length);
for (int i = 0; i < x_error + inverse_two_error; i++) x.RestoreHalfBlock(); for (int i = 0; i < x_error + inverse_two_error; i++) x.RestoreHalfBlock();
/** /**
* now x is the next x, store[tot] stores last x, store[tot^1] stores the x * now x is the next x, store[tot] stores last x, store[tot^1] stores the x