main function

void main()

Implementation

void main() {
  print('========================================');
  print('   LAYER SPEED BENCHMARK (CPU ONLY)     ');
  print('========================================\n');

  int iterations = 100;
  int seqLen = 64;
  int dModel = 128;

  SingleHeadAttention attention = SingleHeadAttention(dModel);
  Tensor<Matrix> attentionInput = generateMatrix(seqLen, dModel);

  runBenchmark<Matrix, Matrix>('SingleHeadAttention (Seq: 64, dModel: 128)', attention, attentionInput, iterations);

  print('========================================');
  print('         BENCHMARK COMPLETE             ');
  print('========================================');
}