Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
63 changes: 1 addition & 62 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -1,62 +1 @@
```
# Python
__pycache__/
*.pyc
*.pyo
*.pyd
*.egg-info/

# Build artifacts
build/
dist/
*.so
*.dylib
*.dll

# Dependencies
.venv/
venv/
.env
.env.local
*.env.*

# Editors
.vscode/
.idea/
*.swp
*.swo

# Logs
*.log

# OS
.DS_Store
Thumbs.db

# Testing
.coverage
coverage/
htmlcov/
.pytest_cache/
.mypy_cache/

# Compressed
*.zip
*.gz
*.tar
*.tgz
*.bz2
*.xz
*.7z
*.rar
*.zst
*.lz4
*.lzh
*.cab
*.arj
*.rpm
*.deb
*.Z
*.lz
*.lzo
```
Nothing should be ignored based on the provided file changes. The only modified file is a C++ source file (`cpp/src/buffer_pool.cpp`), which is a source code file and should not be ignored. There are no build artifacts, dependencies, temporary files, or other items in the change list that warrant being added to `.gitignore`.
Binary file added __pycache__/__init__.cpython-312.pyc
Binary file not shown.
Binary file added __pycache__/_version.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/__init__.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/_bridge.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/adaptive_kv.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/affect.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/amp.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/attention.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/autograd_core.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/base.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/buffer_pool.cpython-312.pyc
Binary file not shown.
Binary file not shown.
Binary file added backend/__pycache__/cells.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/compute.cpython-312.pyc
Binary file not shown.
Binary file not shown.
Binary file added backend/__pycache__/conv.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/core.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/faiss.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/fft.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/fnn.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/fnn_chain.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/learning.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/lora.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/memory.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/normalization.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/pipelines.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/pooling.cpython-312.pyc
Binary file not shown.
Binary file not shown.
Binary file added backend/__pycache__/snn.cpython-312.pyc
Binary file not shown.
Binary file added backend/__pycache__/snn_compute.cpython-312.pyc
Binary file not shown.
Binary file not shown.
Binary file added backend/__pycache__/tensor_ops.cpython-312.pyc
Binary file not shown.
Binary file not shown.
Binary file not shown.
14 changes: 11 additions & 3 deletions cpp/src/buffer_pool.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -115,7 +115,10 @@ GrillyBuffer BufferPool::allocateBuffer(size_t bucketSize) {
VkBufferCreateInfo bufferInfo{};
bufferInfo.sType = VK_STRUCTURE_TYPE_BUFFER_CREATE_INFO;
bufferInfo.size = bucketSize;
bufferInfo.usage = VK_BUFFER_USAGE_STORAGE_BUFFER_BIT;
// Add TRANSFER_DST and TRANSFER_SRC for staging copies (upload/download)
bufferInfo.usage = VK_BUFFER_USAGE_STORAGE_BUFFER_BIT |
VK_BUFFER_USAGE_TRANSFER_DST_BIT |
VK_BUFFER_USAGE_TRANSFER_SRC_BIT;
bufferInfo.sharingMode = VK_SHARING_MODE_EXCLUSIVE;

// REQUIRE device-local memory.
Expand Down Expand Up @@ -276,7 +279,10 @@ GrillyBuffer BufferPool::acquirePreferDeviceLocal(size_t size) {
VkBufferCreateInfo bufferInfo{};
bufferInfo.sType = VK_STRUCTURE_TYPE_BUFFER_CREATE_INFO;
bufferInfo.size = bucket;
bufferInfo.usage = VK_BUFFER_USAGE_STORAGE_BUFFER_BIT;
// Add TRANSFER_DST and TRANSFER_SRC for staging copies (upload/download)
bufferInfo.usage = VK_BUFFER_USAGE_STORAGE_BUFFER_BIT |
VK_BUFFER_USAGE_TRANSFER_DST_BIT |
VK_BUFFER_USAGE_TRANSFER_SRC_BIT;
bufferInfo.sharingMode = VK_SHARING_MODE_EXCLUSIVE;

// Request host-visible mapping but PREFER device-local.
Expand Down Expand Up @@ -323,8 +329,10 @@ GrillyBuffer BufferPool::acquireReadback(size_t size) {
VkBufferCreateInfo bufferInfo{};
bufferInfo.sType = VK_STRUCTURE_TYPE_BUFFER_CREATE_INFO;
bufferInfo.size = bucket;
// Add TRANSFER_SRC for readback (GPU → CPU copy source)
bufferInfo.usage = VK_BUFFER_USAGE_STORAGE_BUFFER_BIT |
VK_BUFFER_USAGE_TRANSFER_DST_BIT;
VK_BUFFER_USAGE_TRANSFER_DST_BIT |
VK_BUFFER_USAGE_TRANSFER_SRC_BIT;
bufferInfo.sharingMode = VK_SHARING_MODE_EXCLUSIVE;

VmaAllocationCreateInfo allocInfo{};
Expand Down
Binary file added experimental/__pycache__/__init__.cpython-312.pyc
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file added experimental/vsa/__pycache__/ops.cpython-312.pyc
Binary file not shown.
Binary file not shown.
Binary file added functional/__pycache__/__init__.cpython-312.pyc
Binary file not shown.
Binary file added functional/__pycache__/_helpers.cpython-312.pyc
Binary file not shown.
Binary file not shown.
Binary file added functional/__pycache__/attention.cpython-312.pyc
Binary file not shown.
Binary file added functional/__pycache__/bridge.cpython-312.pyc
Binary file not shown.
Binary file added functional/__pycache__/cells.cpython-312.pyc
Binary file not shown.
Binary file added functional/__pycache__/dropout.cpython-312.pyc
Binary file not shown.
Binary file added functional/__pycache__/embedding.cpython-312.pyc
Binary file not shown.
Binary file added functional/__pycache__/faiss.cpython-312.pyc
Binary file not shown.
Binary file added functional/__pycache__/fft.cpython-312.pyc
Binary file not shown.
Binary file not shown.
Binary file added functional/__pycache__/linear.cpython-312.pyc
Binary file not shown.
Binary file added functional/__pycache__/loss.cpython-312.pyc
Binary file not shown.
Binary file added functional/__pycache__/memory.cpython-312.pyc
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file added functional/__pycache__/snn.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/__init__.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/_helpers.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/_perf_policy.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/activations.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/addition_linear.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/affect.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/attention.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/autograd.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/capsule.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/capsule_embedding.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/cells.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/containers.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/conv.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/decoding.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/dropout.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/embedding.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/functional.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/gpu_backward.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/hippocampal.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/init.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/linear.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/lora.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/loss.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/memory.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/module.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/module_list.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/modules.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/multimodal.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/normalization.cpython-312.pyc
Binary file not shown.
Binary file not shown.
Binary file added nn/__pycache__/parameter.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/pooling.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/prefix_scan.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/projection_heads.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/rnn.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/routing.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/snn.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/snn_ann2snn.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/snn_attention.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/snn_base.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/snn_containers.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/snn_monitor.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/snn_neurons.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/snn_normalization.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/snn_surrogate.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/snn_synapses.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/transformer.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/utils.cpython-312.pyc
Binary file not shown.
Binary file added nn/__pycache__/vsa_lm.cpython-312.pyc
Binary file not shown.
Binary file added optim/__pycache__/__init__.cpython-312.pyc
Binary file not shown.
Binary file added optim/__pycache__/adam.cpython-312.pyc
Binary file not shown.
Binary file added optim/__pycache__/adamw.cpython-312.pyc
Binary file not shown.
Binary file added optim/__pycache__/base.cpython-312.pyc
Binary file not shown.
Binary file added optim/__pycache__/hypergradient.cpython-312.pyc
Binary file not shown.
Binary file added optim/__pycache__/lr_scheduler.cpython-312.pyc
Binary file not shown.
Binary file not shown.
Binary file added optim/__pycache__/nlms.cpython-312.pyc
Binary file not shown.
Binary file added optim/__pycache__/sgd.cpython-312.pyc
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file added torch_api/__pycache__/__init__.cpython-312.pyc
Binary file not shown.
Binary file added torch_api/__pycache__/amp_mod.cpython-312.pyc
Binary file not shown.
Binary file not shown.
Binary file added torch_api/__pycache__/functional.cpython-312.pyc
Binary file not shown.
Binary file added torch_api/__pycache__/ops.cpython-312.pyc
Binary file not shown.
Binary file not shown.
Binary file added torch_api/__pycache__/tensor.cpython-312.pyc
Binary file not shown.
Binary file added torch_api/__pycache__/vulkan_mod.cpython-312.pyc
Binary file not shown.
Binary file added utils/__pycache__/__init__.cpython-312.pyc
Binary file not shown.
Binary file added utils/__pycache__/checkpoint.cpython-312.pyc
Binary file not shown.
Binary file added utils/__pycache__/data.cpython-312.pyc
Binary file not shown.
Binary file added utils/__pycache__/device.cpython-312.pyc
Binary file not shown.
Binary file added utils/__pycache__/device_manager.cpython-312.pyc
Binary file not shown.
Binary file added utils/__pycache__/grl_checkpoint.cpython-312.pyc
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file added utils/__pycache__/initialization.cpython-312.pyc
Binary file not shown.
Binary file added utils/__pycache__/numba_ops.cpython-312.pyc
Binary file not shown.
Binary file added utils/__pycache__/onnx_exporter.cpython-312.pyc
Binary file not shown.
Binary file added utils/__pycache__/pytorch_compat.cpython-312.pyc
Binary file not shown.
Binary file added utils/__pycache__/pytorch_ops.cpython-312.pyc
Binary file not shown.
Binary file added utils/__pycache__/stable_hash.cpython-312.pyc
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file added utils/__pycache__/visualization.cpython-312.pyc
Binary file not shown.
Binary file added utils/__pycache__/visualizer.cpython-312.pyc
Binary file not shown.
Binary file not shown.
Loading