AtHeartEngineer/icicle
1
0
Fork 0
mirror of https://github.com/pseXperiments/icicle.git synced 2026-01-08 23:17:54 -05:00
Code Issues Packages Projects Releases Wiki Activity

complied with reviewer's comments

Browse Source
This commit is contained in:
stas
2024-02-19 15:59:49 -05:00
parent fb9e5c8511
commit 4664cfded5
2 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/c++/ntt/README.md
View File

@@ -34,5 +34,5 @@ NTT<S, E>(input, ntt_size, NTTDir::kForward, config, output);
1. Define the size of the example
2. Initialize input
3. Run Radix2 NTT
4. Run MexedRadix NTT
4. Run MixedRadix NTT
5. Validate the data output

5
examples/c++/ntt/example.cu
View File

@@ -68,7 +68,7 @@ int main(int argc, char* argv[])
{
std::cout << "Icicle Examples: Number Theoretical Transform (NTT)" << std::endl;
std::cout << "Example parameters" << std::endl;
const unsigned log_ntt_size = 26;
const unsigned log_ntt_size = 20;
std::cout << "Log2(NTT size): " << log_ntt_size << std::endl;
const unsigned ntt_size = 1 << log_ntt_size;
std::cout << "NTT size: " << ntt_size << std::endl;
@@ -96,6 +96,9 @@ int main(int argc, char* argv[])
cudaError_t err = NTT<S, E>(input, ntt_size, NTTDir::kForward, config, output);
END_TIMER(MixedRadix, "MixedRadix NTT");
std::cout << "Validating output" << std::endl;
validate_output(ntt_size, nof_ntts, output);
config.ntt_algorithm = NttAlgorithm::Radix2;
START_TIMER(Radix2);
err = NTT<S, E>(input, ntt_size, NTTDir::kForward, config, output);