def BU(A, B, P): #radix-2 butterfly unit
tmp_ay = np.zeros((1, 2), dtype=int) #tmp_array
tmp_a = Mod.AddMod(A, B, P)
tmp_b = Mod.SubMod(A, B, P)
tmp_ay[0][0] = tmp_a #a+b
tmp_ay[0][1] = tmp_b #a-b
return tmp_ay
def BU(A, B, P): #radix-2 butterfly unit
tmp_ay = np.zeros((1, 2), dtype=int) #tmp_array
tmp_a = Mod.AddMod(A, B, P)
tmp_b = Mod.SubMod(A, B, P)
tmp_ay[0][0] = tmp_a #a+b
tmp_ay[0][1] = tmp_b #a-b
return tmp_ay
