| .. |
|
fused_softmax.cpp
|
ed553e9238
Add Megatron attention implementation for benchmarking
|
hace 2 años |
|
scaled_masked_softmax.h
|
ed553e9238
Add Megatron attention implementation for benchmarking
|
hace 2 años |
|
scaled_masked_softmax_cuda.cu
|
ed553e9238
Add Megatron attention implementation for benchmarking
|
hace 2 años |
|
scaled_upper_triang_masked_softmax.h
|
ed553e9238
Add Megatron attention implementation for benchmarking
|
hace 2 años |
|
scaled_upper_triang_masked_softmax_cuda.cu
|
ed553e9238
Add Megatron attention implementation for benchmarking
|
hace 2 años |
|
setup.py
|
50896ec574
Make nvcc threads configurable via environment variable (#885)
|
hace 9 meses |
|
type_shim.h
|
ed553e9238
Add Megatron attention implementation for benchmarking
|
hace 2 años |
Chirag Jain
50896ec574
Make nvcc threads configurable via environment variable (#885)