You are viewing a plain text version of this content. The canonical link for it is here.
Posted to dev@mahout.apache.org by "ryankert01 (via GitHub)" <gi...@apache.org> on 2026/01/20 10:13:00 UTC

Re: [PR] MAHOUT-878 Add CUDA Torch Tensor Support for QDP Python Binding [mahout]

ryankert01 commented on PR #881:
URL: https://github.com/apache/mahout/pull/881#issuecomment-3772075605

   ```
     ⎿  ============================================================                                                                         
        CUDA Tensor Encoding Speed Comparison                                                                                                
        ============================================================                                                                         
        Batch=  100, Features=  64 | CPU:    0.30ms | CUDA:    0.12ms | Speedup: 2.45x                                                       
        Batch=  100, Features= 256 | CPU:    0.21ms | CUDA:    0.46ms | Speedup: 0.45x                                                       
        Batch=  100, Features=1024 | CPU:    0.67ms | CUDA:    0.17ms | Speedup: 3.94x                                                       
        Batch= 1000, Features=  64 | CPU:    3.82ms | CUDA:    0.48ms | Speedup: 7.93x                                                       
        Batch= 1000, Features= 256 | CPU:    2.90ms | CUDA:    0.26ms | Speedup: 11.18x                                                      
        Batch= 1000, Features=1024 | CPU:    6.11ms | CUDA:    0.57ms | Speedup: 10.78x                                                      
        Batch=10000, Features=  64 | CPU:    4.26ms | CUDA:    2.03ms | Speedup: 2.10x                                                       
        Batch=10000, Features= 256 | CPU:   10.09ms | CUDA:    4.78ms | Speedup: 2.11x                                                       
        Batch=10000, Features=1024 | CPU:   27.10ms | CUDA:   10.42ms | Speedup: 2.60x                                                       
        ============================================================   
   ```


-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: dev-unsubscribe@mahout.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org