Skip to content

fix: memory_bytes dtype bug, remove debug logging, cleanup temp files,auto-detect GPUs#9

Open
kunwar-vikrant wants to merge 1 commit into
0xSero:mainfrom
kunwar-vikrant:fix/cleanup-and-bugfixes
Open

fix: memory_bytes dtype bug, remove debug logging, cleanup temp files,auto-detect GPUs#9
kunwar-vikrant wants to merge 1 commit into
0xSero:mainfrom
kunwar-vikrant:fix/cleanup-and-bugfixes

Conversation

@kunwar-vikrant
Copy link
Copy Markdown

@kunwar-vikrant kunwar-vikrant commented Apr 4, 2026

  • store.py: memory_bytes() now uses tensor.element_size() instead of hardcoded *2, fixing underestimation for float32 tensors
  • vllm_attn_backend.py: replaced /tmp/tq_debug.log file writes with proper logger.debug()/logger.error() calls
  • benchmark.py, proof.py: added try/finally to clean up /tmp/tq_*.py temp files after subprocess runs
  • benchmark.py, proof.py: replaced hardcoded GPU fallback '0,1,4,6' with _default_gpu_ids() auto-detection via torch.cuda.device_count()
  • tests/test_fixes.py: added 8 tests covering all changes

@kunwar-vikrant kunwar-vikrant changed the title fix: memory_bytes dtype bug, remove debug logging, cleanup temp files… fix: memory_bytes dtype bug, remove debug logging, cleanup temp files,auto-detect GPUs Apr 4, 2026
…, auto-detect GPUs

- store.py: memory_bytes() now uses tensor.element_size() instead of
  hardcoded *2, fixing underestimation for float32 tensors
- vllm_attn_backend.py: replaced /tmp/tq_debug.log file writes with
  proper logger.debug()/logger.error() calls
- benchmark.py, proof.py: added try/finally to clean up /tmp/tq_*.py
  temp files after subprocess runs
- benchmark.py, proof.py: replaced hardcoded GPU fallback '0,1,4,6'
  with _default_gpu_ids() auto-detection via torch.cuda.device_count()
@kunwar-vikrant kunwar-vikrant force-pushed the fix/cleanup-and-bugfixes branch from 5828620 to 3ff1c8e Compare April 4, 2026 12:42
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant