/stm-benchmark

Primary LanguageScalaApache License 2.0Apache-2.0

STM benchmarks

Benchmarks for software transactional memory (STM) implementations in Scala

Based on the idea of chrisseaton/ruby-stm-lee-demo (and originally on Lee-TM), we've implemented (a simplified version of) Lee’s routing algorithm, and used STM to parallelize it.

Further reading:

  • https://chrisseaton.com/truffleruby/ruby-stm/ (the Ruby implementation referred to above),
  • and the two papers about Lee-TM:
    • Ian Watson, Chris Kirkham and Mikel Luján. "A Study of a Transactional Parallel Routing Algorithm." In Proceedings of the 16th International Conference on Parallel Architectures and Compilation Techniques (PACT 2007), Brasov, Romania, Sept. 2007, pp 388-398. (PDF)
    • Mohammad Ansari, Christos Kotselidis, Kim Jarvis, Mikel Luján, Chris Kirkham, and Ian Watson. "Lee-TM: A Non-trivial Benchmark for Transactional Memory." In Proceedings of the 8th International Conference on Algorithms and Architectures for Parallel Processing (ICA3PP 2008), Aiya Napa, Cyprus, June 2008. (PDF)

Tested STM implementations

We've implemented Lee's algorithm with various Scala STMs. We've tried to implement the algorithm as similar as reasonably possible in every implementation, but we didn't write (intentionally) unidiomatic code just to be more similar. The tested/measured STMs are as follows (with some remarks for each implementation):

  • Cats STM in folder cats-stm.
    • We run the Cats STM transactions on a Cats Effect runtime, which they're designed to run on.
    • We disable tracing in the runtime, to avoid the negative performance impact.
    • Cats STM doesn't have a built-in TArray or similar type, so we use Array[TVar[A]] for the board matrices.
  • CHOAM in folder choam
    • This is technically not an STM, but close enough (this algorithm doesn't require everything from an STM, e.g., there is no need for the orElse combinator).
    • We run the Rxns on a Cats Effect runtime, which they're designed to run on.
    • We disable tracing in the runtime, to avoid the negative performance impact.
    • For the board matrices we use the built-in Ref.array in CHOAM.
  • ScalaSTM in folder scala-stm
    • We've implemented 2 versions: ScalaStmSolver uses the ScalaSTM API in an idiomatic way, while WrStmSolver wraps the ScalaSTM API in a monadic API similar to that of Cats STM or ZSTM. This way we can also get some ideas about the overhead of a monadic ("programs as values") API.
    • For easy parallelization, we run the ScalaSTM transactions on a Cats Effect runtime. ScalaSTM sometimes blocks threads, but does this by using scala.concurrent.BlockContext, which is supported by the Cats Effect runtime (it starts compensating threads as necessary), so this should be fine (although not ideal; but performance seems fine).
    • We disable tracing in the runtime, to avoid the negative performance impact.
    • We use ScalaSTM's TArray for the board matrices.
  • ZSTM in folder zstm.
    • We run the ZSTM transactions on their own zio.Runtime, which they seem designed for.
    • We disable FiberRoots in the runtime, to avoid the negative performance impact.
    • We use ZSTM's TArray for the board matrices.

Some general remarks:

  • The transactions in these implementations of Lee’s routing algorithm are read heavy, but at the end they always write to some locations (to lay a route). This means that read-only transactions, and transactions which only access a very small number of TVars are not tested/measured.
  • We also have a (baseline) sequential (non-parallelized) implementation of the same algorithm in folder sequential. This sequential implementation is intentionally not very well optimized, because we'd like to compare it to similarly high-level and easy to use STMs.

Benchmarks

Benchmarks are in Benchmarks.scala. They can be configured with the following JMH parameters:

  • board (String): the input(s) are specified by this parameter, which is a filename to be loaded from classpath resources.
  • seed (Long): before solving, the boards are "normalized" with a pseudorandom shuffle; this is the random seed to use.
  • restrict (Int): before solving, the boards are "restricted", i.e., some of the routes are removed from them. This makes solving them easier (because there is less work, and also less change of conflicts). The value passed to this parameter will be used to >> (right shift) the number of routes; e.g., restrict=1 will remove approx. half of the routes. (The routes to remove are chosen pseudorandomly based on seed.)

The various parallel implementations are tunable with more parameters:

  • parLimit (Int): parallelism is limited to this value (e.g., with parTraverseN); specify 0 to use Runtime.getRuntime().availableProcessors().
  • txnLimitMultiplier (Int; Cats STM only): we pass txnLimitMultiplier * parLimit to Cats STM as the number of transactions that are allowed to evaluate concurrently (the n argument to STM.runtime).
  • strategy (String; CHOAM only): the Rxn.Strategy to use for backoff (spin | cede | sleep).