Homework 3. Java shared memory performance races

Background

You're working for a startup company Ginormous Data Inc. (GDI) that specializes in finding patterns in large amounts of data. For example, a big retailer might give GDI all the web visits and purchases made and all the credit reports they've inspected and records of all the phone calls to them, and GDI will then find patterns in the data that suggest which toys will be hot this Christmas season. The programs that GDI writes are mostly written in Java. They aren't perfect; they're just heuristics, and they're operating on incomplete and sometimes-inaccurate information. They do need to be fast, though, as your clients are trying to find patterns faster than their competition can, and are willing to put up with a few errors even if the results aren't perfect, so long as they get good-enough results quickly.

The problem

GDI regularly uses multithreading to speed up its applications, and many of GDI's programs operate on shared-memory representations of the state of a simulation. These states are updated safely, using Java's synchronized keyword, and this is known to be a bottleneck in the code. Your boss asks you what will happen if you remove the synchronized keyword. You reply, "It'll break the simulations." She responds, "So what? If it's just a small amount of breakage, that might be good enough. Or maybe you can substitute some other synchronization strategy that's less heavyweight, and that'll be good enough." She suggests that you look into this by measuring how often GDI's programs are likely to break if they switch to inadequate-but-faster synchronization methods, and also by looking into a safe technique that is faster than synchronized.

"Oh, and while you're at it," your boss says, "also look into whether we'll get better performance by using the virtual threads delivered in JDK 19 and finalized in JDK 21." You know how bosses are.

In some sense the first part of this assignment is the reverse of what software engineers traditionally do with multithreaded applications. Traditionally, they are worried about race conditions and insert enough synchronization so that the races become impossible. Here, though, you're deliberately trying to add races to the code in order to speed it up, and want to measure whether (and ideally, how badly) things will break if you do.

It should be noted that we are on thin ice here, so thin that some would argue we've gone over the edge. That's OK: we're experimenting! For more about the overall topic, please see: Boehm H-J, Adve SV. You don't know jack about shared variables or memory models. ACM Queue 2011 Dec;9(12):40. doi:10.1145/2076796.2088916.

The Java memory model

Java synchronization is based on the Java memory model (JMM), which defines how an application can safely avoid data races when accessing shared memory. The JMM lets Java implementations optimize accesses by allowing more behaviors than the intuitive semantics where there is a global clock and actions by threads interleave in a schedule that assumes sequential consistency. On modern hardware, these intuitive semantics are often incorrect: for example, intraprocessor cache communication might be faster than memory, which means that a cached read can return a new value on one processor before an uncached read returns an old value on another. To allow this kind of optimization, first, the JMM says that two accesses to the same location conflict if they come from different threads, at least one is a write, and the location is not declared to be volatile; and second, the JMM says that behavior is well-defined to be data-race free (DRF) unless two conflicting accesses occur without synchronization in between.

The details for proving that a program is DRF can be tricky, as is optimizing a Java implementation with data-race freedom in mind. Not only have serious memory-synchronization bugs been found in Java implementations, occasionally bugs have been found in the JMM itself, and sometimes people have even announced bugs only to find out later that they weren't bugs after all. For more details about this, please see: Lochbihler A. Making the Java memory model safe [PDF]. ACM TOPLAS. 2013 Dec;35(4):12. doi:10.1145/2518191. You needn't read all this paper, just the first eight pages or so—through the end of §1.1.3.

How to break sequential consistency in Java

It's easy to write programs that break sequential consistency in Java. To model this, you will use a simple prototype that manages a data structure that represents an array of longs. Each array entry starts at zero. A state transition, called a swap, consists of subtracting 1 from one of the entries, and adding 1 to an entry – typically a different entry although the two entries can be the same in which case a swap does nothing. The sum of all the array entries should therefore remain zero; if it becomes nonzero, that indicates that one or more transitions weren't done correctly. The converse is not true: if the sum is zero it's still possible that some state transitions were done incorrectly. Still, this test is a reasonable way to check for errors in the simulation.

For an example of a simulation, see jmm.jar, a JAR file containing the simplified source code of a simulation. It contains the following interfaces and classes:

State
The API for a simulation state. The only way to change the state is to invoke swap(i,j), where i and j are indexes into the array. This subtracts 1 from the ith entry and adds 1 to the jth entry.
NullState
An implementation of State that does nothing. Swapping has no effect. This is used for timing the scaffolding of the simulation.
SynchronizedState
An implementation of State that uses the Synchronized class so that it is safe but slow.
SwapTest
A Runnable class that tests a state implementation by performing a given number of swap transitions on it.
UnsafeMemory
A test harness, with a main method. Invoke it via a shell command like this:
time timeout 3600 java UnsafeMemory Platform 5 Synchronized 8 100000000
Here, the initial time tells the shell to give top-level real, user and system time. The timeout 3600 puts a 3600-second timeout on Java, helpful if your program mistakenly loops forever. Platform says to use platform threads; 5 says to use a state array of 5 entries; Synchronized means to test the SynchronizedState implementation; 8 means to divide the work into 8 threads of roughly equal size; 100000000 means to do 100 million swap transitions total. The output of this command should look something like this:
Total real time 3.31866 s
Average real swap time 265.493 ns

real    0m3.424s
user    0m3.775s
sys     0m0.105s
The output says that from Java’s point of view the entire test took 3.31866 s in real time. It also says that a single swap by a single thread took an average of 265.493 ns of real time. From the Linux kernel’s point of view, the entire Java program consumed 3.424 s real time, 3.775 s user CPU time, and 0.105 s system CPU time.

Assignment

Build and use a sequential-consistency-violating performance and reliability testing program, along the lines described below.

Do the following tasks and submit work that embodies your results.

  1. Implement a new class UnsynchronizedState, which is implemented just like SynchronizedState except that it does not use the keyword synchronized in its implementation. Put the implementation into a new file UnsynchronizedState.java.
  2. If you like, you can implement further classes for relaxed synchronization and increased performance; see the AcmeSafeState comment in the sample code. This is optional.
  3. Integrate all the classes into a single program UnsafeMemory, which you should be able to compile with the command 'javac *.java' and to run using the same sort of shell command as the test harness. Your version of UnsafeMemory.java will need to differ slightly from the one distributed as part of this assignment
  4. For each class SynchronizedState and UnsynchronizedState, measure and characterize the class's performance and reliability as described above (that is, vary whether platform or virtual threads are used, vary the state array size, and vary the number of threads). Attempt to find areas where these differ.
  5. Compare the classes' reliability and performance to each other. For each class, say what kind of benchmarks (if any) the class seems to do best and worst on.

Write a report that contains the following explanations and discussions. Your report should be at least one and at most two pages long (not counting any appendixes), using 10-point font in a two-column format on an 8½"×11" page, as suggested in the USENIX template mentioned in Resources for written reports and oral presentations.

Submit

Submit two files:

  1. A JAR file jmmplus.jar containing your solution. Include your source code (.java files) not the .class files. It should contain a copy of the files in jmm.jar, possibly with modifications (though you should attempt to minimize these modifications). It should also contain the source code to your new classes. Please limit your source-code lines to 80 characters or less.
  2. A PDF file report.pdf containing your explanations, discussions, and performance and reliability results.

Do not put your name or student-ID into your submissions.