[5adf4f4] | 1 | import akka.actor.typed.scaladsl.Behaviors |
---|
| 2 | import akka.actor.typed.scaladsl.LoggerOps |
---|
| 3 | import akka.actor.typed.{ ActorRef, ActorSystem, Behavior } |
---|
| 4 | import java.util.concurrent.atomic.AtomicInteger |
---|
| 5 | import akka.event.Logging |
---|
| 6 | import akka.actor.typed.DispatcherSelector |
---|
| 7 | import java.util.concurrent.Semaphore |
---|
| 8 | |
---|
| 9 | object MatrixMult { |
---|
| 10 | final case class WorkMsg( var Z: Array[Int], val X: Array[Int], val Y: Array[Array[Int]] ) |
---|
| 11 | |
---|
| 12 | def apply(): Behavior[WorkMsg] = Behaviors.receive { (context, message) => |
---|
| 13 | // message match { |
---|
| 14 | // case WorkMsg( Z, X, Y ) => |
---|
| 15 | for ( i <- 0 until MatrixMain.yc ) { |
---|
| 16 | message.Z(i) = 0 |
---|
| 17 | for ( j <- 0 until MatrixMain.xc ) { |
---|
| 18 | message.Z(i) = message.Z(i) + message.X(j) * message.Y(j)(i) |
---|
| 19 | } |
---|
| 20 | } |
---|
| 21 | if ( MatrixMain.actorCnt.incrementAndGet() == MatrixMain.xr ) { |
---|
| 22 | MatrixMain.system ! MatrixMain.Stop() |
---|
| 23 | } |
---|
| 24 | // } // message |
---|
| 25 | Behaviors.same |
---|
| 26 | } |
---|
| 27 | } |
---|
| 28 | |
---|
| 29 | object MatrixMain { |
---|
| 30 | sealed trait MainMessages |
---|
| 31 | final case class Start() extends MainMessages |
---|
| 32 | final case class Stop() extends MainMessages |
---|
| 33 | |
---|
| 34 | var xr = 3_072; var xc = 3_072; var yc = 3_072; var Processors = 1; var MaxProcs = 48; // default values |
---|
| 35 | |
---|
| 36 | val actors = new Array[ActorRef[MatrixMult.WorkMsg]](xr) |
---|
| 37 | val messages = new Array[MatrixMult.WorkMsg](xr); |
---|
| 38 | |
---|
| 39 | val actorCnt = new AtomicInteger(0) |
---|
| 40 | val sem = new Semaphore(0) |
---|
| 41 | |
---|
| 42 | var system : ActorSystem[MatrixMain.MainMessages] = null |
---|
| 43 | var startTime = System.nanoTime() |
---|
| 44 | |
---|
| 45 | var X = Array.ofDim[Int](0, 0) // set type, set size below |
---|
| 46 | var Y = Array.ofDim[Int](0, 0) |
---|
| 47 | var Z = Array.ofDim[Int](0, 0) |
---|
| 48 | |
---|
| 49 | def apply(): Behavior[MainMessages] = Behaviors.setup { context => |
---|
| 50 | for ( id <- 0 until xr ) { // create actors |
---|
| 51 | actors(id) = context.spawn(MatrixMult(), "actor_" + id, DispatcherSelector.fromConfig("akka.dispatcher")) |
---|
| 52 | messages(id) = new MatrixMult.WorkMsg(Z(id), X(id), Y) |
---|
| 53 | } // for |
---|
| 54 | |
---|
| 55 | Behaviors.receiveMessage { message => |
---|
| 56 | message match { |
---|
| 57 | case Start() => |
---|
| 58 | for ( id <- 0 until xr ) { // start actors |
---|
| 59 | actors(id) ! messages(id) |
---|
| 60 | } // for |
---|
| 61 | Behaviors.same |
---|
| 62 | case Stop() => |
---|
| 63 | sem.release() |
---|
| 64 | Behaviors.stopped |
---|
| 65 | } // match |
---|
| 66 | } |
---|
| 67 | } |
---|
| 68 | |
---|
| 69 | def usage() = { |
---|
| 70 | println( "Usage: " + |
---|
| 71 | s"[ yc (> 0) | 'd' (default ${xr}) ] " + |
---|
| 72 | s"[ xc (> 0) | 'd' (default ${xc}) ] " + |
---|
| 73 | s"[ xr (> 0) | 'd' (default ${yc}) ] " + |
---|
| 74 | s"[ processors (> 0) | 'd' (default ${Processors}) ]" |
---|
| 75 | ) |
---|
| 76 | System.exit( 1 ) |
---|
| 77 | } |
---|
| 78 | |
---|
| 79 | def main(args: Array[String]): Unit = { |
---|
| 80 | if ( args.length > 4 ) usage() // process command-line arguments |
---|
| 81 | if ( args.length == 5 ) { |
---|
| 82 | if ( args(4) != "d" ) { // default ? |
---|
| 83 | MaxProcs = args(4).toInt |
---|
| 84 | if ( MaxProcs < 1 ) usage() |
---|
| 85 | } // if |
---|
| 86 | } // if |
---|
| 87 | if ( args.length >= 4 ) { |
---|
| 88 | if ( args(3) != "d" ) { // default ? |
---|
| 89 | Processors = args(3).toInt |
---|
| 90 | if ( Processors < 1 ) usage() |
---|
| 91 | } // if |
---|
| 92 | } // if |
---|
| 93 | if ( args.length >= 3 ) { // fall through |
---|
| 94 | if ( args(2) != "d" ) { // default ? |
---|
| 95 | xr = args(2).toInt |
---|
| 96 | if ( xr < 1 ) usage() |
---|
| 97 | } // if |
---|
| 98 | } // if |
---|
| 99 | if ( args.length >= 2 ) { // fall through |
---|
| 100 | if ( args(1) != "d" ) { // default ? |
---|
| 101 | xc = args(1).toInt |
---|
| 102 | if ( xc < 1 ) usage() |
---|
| 103 | } // if |
---|
| 104 | } // if |
---|
| 105 | if ( args.length >= 1 ) { // fall through |
---|
| 106 | if ( args(0) != "d" ) { // default ? |
---|
| 107 | yc = args(0).toInt |
---|
| 108 | if ( yc < 1 ) usage() |
---|
| 109 | } // if |
---|
| 110 | } // if |
---|
| 111 | |
---|
| 112 | xr = xr / (MaxProcs / Processors) |
---|
| 113 | |
---|
| 114 | X = Array.ofDim[Int](xr, xc) |
---|
| 115 | Y = Array.ofDim[Int](xc, yc) |
---|
| 116 | Z = Array.ofDim[Int](xr, yc) |
---|
| 117 | |
---|
| 118 | for ( r <- 0 until xr ) { |
---|
| 119 | for ( c <- 0 until xc ) { |
---|
| 120 | X(r)(c) = r * c % 37 |
---|
| 121 | } |
---|
| 122 | } |
---|
| 123 | |
---|
| 124 | for ( r <- 0 until xc ) { |
---|
| 125 | for ( c <- 0 until yc ) { |
---|
| 126 | Y(r)(c) = r * c % 37 |
---|
| 127 | } |
---|
| 128 | } |
---|
| 129 | |
---|
| 130 | startTime = System.nanoTime() |
---|
| 131 | system = ActorSystem( MatrixMain(), "Matrix" ) |
---|
| 132 | system ! Start() |
---|
| 133 | |
---|
| 134 | sem.acquire() |
---|
| 135 | |
---|
| 136 | println( s"${Processors}\t" + f"${(System.nanoTime() - startTime) * (MaxProcs / Processors) / 1_000_000_000.0}%1.2f" ) |
---|
| 137 | } |
---|
| 138 | } |
---|
| 139 | |
---|
| 140 | // Local Variables: // |
---|
| 141 | // tab-width: 4 // |
---|
| 142 | // mode: java // |
---|
| 143 | // compile-command: "sbt --warn -J-Xmx32g \"run\"" // |
---|
| 144 | // End: // |
---|