Repository navigation

[WIP] GPU support in PEPSKit - #407

Draft

kshyatt wants to merge 74 commits into

mainfrom

ksh/cupeps

Draft

[WIP] GPU support in PEPSKit#407
kshyatt wants to merge 74 commits into
mainfrom
ksh/cupeps

Commits on Oct 22, 2026
- Initial steps to CUDA support
  kshyatt
  committed
  5852f8d
- Restore singular value distance
  kshyatt
  committed
  6bdcf9d
- Cleanup and some actual tests
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  b6412c4
- Try densitymatrices tests
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  cb89560
- Add a simple timestep example
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  6ac1e4f
- More progress
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  a41c960
- refactor `_fliptwist_s`
  
  lkdvos
  authored and
  kshyatt
  committed
  b342a06
- Working CTMRG contractions tests
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  5b0deb9
- Partially support cluster projectors tests
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  1b006d4
- Working tf_ising_finiteT
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  8481d83
- Sitedep truncation also working
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  b0147e0
- Working ctmrg/partition_function (except C4v)
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  41bc277
- Working BP tests
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  f226b78
- Fixed iteration appears to be working
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  fccd717
- Flavors working too
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  a1dfaa7
- Working ctmrg/unitcell
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  f48bfa1
- Working CTMRG/pepo
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  7bc34a0
- Try simplifying the test script
  kshyatt
  committed
  f5283ae
- Try splitting groups
  kshyatt
  committed
  f053625
- Include in title
  kshyatt
  committed
  9a2adb1
- Adjust paths
  kshyatt
  committed
  8a54fde
- Fix runtests.jl
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  ec335e0
- Soothe VUMPS on 1.12
  kshyatt
  committed
  d1a7a41
- Bondenv tests
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  5f8c7ed
- Fix product_peps
  kshyatt
  committed
  1ad5262
- BP-Gaugefix test for CUDA
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  90bd815
- More working tests with downstream fixes
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  d416116
- Don't use QRIteration for C4V + CUDA
  kshyatt
  committed
  e941b79
- Get cuda flavors working
  kshyatt
  committed
  d89a278
- Make the gaugefix loop GPU-friendly
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  70ce884
- CUDA expvals test works
  kshyatt
  committed
  6940c78
- bp/expvals tests working
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  a7b455d
- GPUified SVD and passing tests
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  ae35660
- Working eigh
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  9d6aa46
- Switch to CUDACore and more tests working
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  52dc9be
- Gradient tests
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  ae31ae6
- Fix for c4v
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  4a47032
- Use the new braiding branch over at TK
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  d97c038
- Use alloc_caches from GPUArrays
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  f940238
- Leave a note about which GPU algo to use
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  414b901
- Use DivideAndConquer for ROCm tests
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  27bbeb7
- Let the cache depth vary by algo
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  a98794a
- Fix AMDGPU rev
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  aff3444
- CUDA/CUDACore badness STRIKES AGAIN
  kshyatt
  committed
  b15eea9
- Make sure buffers aren't overwritten in rrule_via_ad
  kshyatt
  committed
  15d2492
- use Jacobi for the fixed_iterscheme tests for speeeeeeeed
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  c158056
- Fixes for holding cache memory for too long
  kshyatt
  committed
  f43933c
- Try turning off the most memory intensive part
  kshyatt
  committed
  a0d1c10
- Add a quick check to make sure no empty spaces were created
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  ca3ce8b
- Batched SVD support and more caching improvements
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  fc5ba12
- Force real-valued accumulator
  kshyatt
  committed
  b01a990
- Move batching logic into MAK as much as possible
  kshyatt
  committed
  3ce3868
- Incremental trimming of duplicated tests
  kshyatt
  committed
  9a2c6e3
- Use new TO version
  kshyatt
  committed
  db8ac54
- Use backend and allocator for the VUMPS transfer matrices
  kshyatt
  committed
  7d6d690
- Various ctmrg fixes
  kshyatt
  committed
  af55bf4
- Fix the c4v gradients
  kshyatt
  committed
  5ddb95f
- Update caching strategy
  kshyatt
  committed
  4e7933d
- Trim down the GPU tests so they run faster on CI
  kshyatt
  committed
  a986c77
- Don't run a test on AMDGPU that needs geev
  kshyatt
  committed
  fba6fcc
- Soothe CI a bit more
  kshyatt
  committed
  75c4cc1
- Try GPUArrays 12
  kshyatt
  committed
  6165465
- Fixups for Project files
  
  Katharine Hyatt
  authored and
  kshyatt
  committed
  db78c2b
- Drop some useless changes
  kshyatt
  committed
  065bf4d
- Remove duplicated storage type methods
  kshyatt
  committed
  6f71a51
- Try the handle handler fix branch
  kshyatt
  committed
  535aa7f
- Update Project.toml
  kshyatt
  committed
  f9b95f9
- Cleanup some rebase dupes
  kshyatt
  committed
  bf8db6a
- Rebase issues and other fixes
  kshyatt
  committed
  862bcf9
- Cleanup some dead lines
  kshyatt
  committed
  6837f90
- Remove more dead lines
  kshyatt
  committed
  a964f3d
- More dead stuff removal
  kshyatt
  committed
  8a20c1b
- Restore minimal kwargs for some tests
  kshyatt
  committed
  717a06c
- GPU specific extensions for SVD stuff
  kshyatt
  committed
  57b667c

Commits

Commits on Oct 22, 2026