| ... | ... | @@ -296,15 +296,11 @@ test "tryAllocReg: no spilling" { |
| 296 | 296 | }; |
| 297 | 297 | defer function.deinit(); |
| 298 | 298 | |
| 299 | | var mock_instruction = ir.Inst{ |
| 300 | | .tag = .breakpoint, |
| 301 | | .ty = Type.initTag(.void), |
| 302 | | .src = .unneeded, |
| 303 | | }; |
| 299 | const mock_instruction: Air.Inst.Index = 1; |
| 304 | 300 | |
| 305 | | try expectEqual(@as(?MockRegister1, .r2), function.register_manager.tryAllocReg(&mock_instruction, &.{})); |
| 306 | | try expectEqual(@as(?MockRegister1, .r3), function.register_manager.tryAllocReg(&mock_instruction, &.{})); |
| 307 | | try expectEqual(@as(?MockRegister1, null), function.register_manager.tryAllocReg(&mock_instruction, &.{})); |
| 301 | try expectEqual(@as(?MockRegister1, .r2), function.register_manager.tryAllocReg(mock_instruction, &.{})); |
| 302 | try expectEqual(@as(?MockRegister1, .r3), function.register_manager.tryAllocReg(mock_instruction, &.{})); |
| 303 | try expectEqual(@as(?MockRegister1, null), function.register_manager.tryAllocReg(mock_instruction, &.{})); |
| 308 | 304 | |
| 309 | 305 | try expect(function.register_manager.isRegAllocated(.r2)); |
| 310 | 306 | try expect(function.register_manager.isRegAllocated(.r3)); |
| ... | ... | @@ -328,28 +324,24 @@ test "allocReg: spilling" { |
| 328 | 324 | }; |
| 329 | 325 | defer function.deinit(); |
| 330 | 326 | |
| 331 | | var mock_instruction = ir.Inst{ |
| 332 | | .tag = .breakpoint, |
| 333 | | .ty = Type.initTag(.void), |
| 334 | | .src = .unneeded, |
| 335 | | }; |
| 327 | const mock_instruction: Air.Inst.Index = 1; |
| 336 | 328 | |
| 337 | | try expectEqual(@as(?MockRegister1, .r2), try function.register_manager.allocReg(&mock_instruction, &.{})); |
| 338 | | try expectEqual(@as(?MockRegister1, .r3), try function.register_manager.allocReg(&mock_instruction, &.{})); |
| 329 | try expectEqual(@as(?MockRegister1, .r2), try function.register_manager.allocReg(mock_instruction, &.{})); |
| 330 | try expectEqual(@as(?MockRegister1, .r3), try function.register_manager.allocReg(mock_instruction, &.{})); |
| 339 | 331 | |
| 340 | 332 | // Spill a register |
| 341 | | try expectEqual(@as(?MockRegister1, .r2), try function.register_manager.allocReg(&mock_instruction, &.{})); |
| 333 | try expectEqual(@as(?MockRegister1, .r2), try function.register_manager.allocReg(mock_instruction, &.{})); |
| 342 | 334 | try expectEqualSlices(MockRegister1, &[_]MockRegister1{.r2}, function.spilled.items); |
| 343 | 335 | |
| 344 | 336 | // No spilling necessary |
| 345 | 337 | function.register_manager.freeReg(.r3); |
| 346 | | try expectEqual(@as(?MockRegister1, .r3), try function.register_manager.allocReg(&mock_instruction, &.{})); |
| 338 | try expectEqual(@as(?MockRegister1, .r3), try function.register_manager.allocReg(mock_instruction, &.{})); |
| 347 | 339 | try expectEqualSlices(MockRegister1, &[_]MockRegister1{.r2}, function.spilled.items); |
| 348 | 340 | |
| 349 | 341 | // Exceptions |
| 350 | 342 | function.register_manager.freeReg(.r2); |
| 351 | 343 | function.register_manager.freeReg(.r3); |
| 352 | | try expectEqual(@as(?MockRegister1, .r3), try function.register_manager.allocReg(&mock_instruction, &.{.r2})); |
| 344 | try expectEqual(@as(?MockRegister1, .r3), try function.register_manager.allocReg(mock_instruction, &.{.r2})); |
| 353 | 345 | } |
| 354 | 346 | |
| 355 | 347 | test "tryAllocRegs" { |
| ... | ... | @@ -377,16 +369,12 @@ test "allocRegs" { |
| 377 | 369 | }; |
| 378 | 370 | defer function.deinit(); |
| 379 | 371 | |
| 380 | | var mock_instruction = ir.Inst{ |
| 381 | | .tag = .breakpoint, |
| 382 | | .ty = Type.initTag(.void), |
| 383 | | .src = .unneeded, |
| 384 | | }; |
| 372 | const mock_instruction: Air.Inst.Index = 1; |
| 385 | 373 | |
| 386 | 374 | try expectEqual([_]MockRegister2{ .r0, .r1, .r2 }, try function.register_manager.allocRegs(3, .{ |
| 387 | | &mock_instruction, |
| 388 | | &mock_instruction, |
| 389 | | &mock_instruction, |
| 375 | mock_instruction, |
| 376 | mock_instruction, |
| 377 | mock_instruction, |
| 390 | 378 | }, &.{})); |
| 391 | 379 | |
| 392 | 380 | // Exceptions |
| ... | ... | @@ -402,13 +390,9 @@ test "getReg" { |
| 402 | 390 | }; |
| 403 | 391 | defer function.deinit(); |
| 404 | 392 | |
| 405 | | var mock_instruction = ir.Inst{ |
| 406 | | .tag = .breakpoint, |
| 407 | | .ty = Type.initTag(.void), |
| 408 | | .src = .unneeded, |
| 409 | | }; |
| 393 | const mock_instruction: Air.Inst.Index = 1; |
| 410 | 394 | |
| 411 | | try function.register_manager.getReg(.r3, &mock_instruction); |
| 395 | try function.register_manager.getReg(.r3, mock_instruction); |
| 412 | 396 | |
| 413 | 397 | try expect(!function.register_manager.isRegAllocated(.r2)); |
| 414 | 398 | try expect(function.register_manager.isRegAllocated(.r3)); |
| ... | ... | @@ -416,7 +400,7 @@ test "getReg" { |
| 416 | 400 | try expect(!function.register_manager.isRegFree(.r3)); |
| 417 | 401 | |
| 418 | 402 | // Spill r3 |
| 419 | | try function.register_manager.getReg(.r3, &mock_instruction); |
| 403 | try function.register_manager.getReg(.r3, mock_instruction); |
| 420 | 404 | |
| 421 | 405 | try expect(!function.register_manager.isRegAllocated(.r2)); |
| 422 | 406 | try expect(function.register_manager.isRegAllocated(.r3)); |