shisa-ai
/

megablocks-hip

Model card Files Files and versions

megablocks-hip / build /torch28-cxx11-rocm64-x86_64-linux /megablocks /ops

42.8 kB

3 contributors

History: 1 commit

leonardlin's picture

Add ROCm build artifacts and HIP backend

1e407f0 about 2 months ago

__init__.py

826 Bytes

Add ROCm build artifacts and HIP backend about 2 months ago
all_to_all_benchmark.py

1.56 kB

Add ROCm build artifacts and HIP backend about 2 months ago
all_to_all_benchmark.sh

252 Bytes

Add ROCm build artifacts and HIP backend about 2 months ago
binned_gather.py

945 Bytes

Add ROCm build artifacts and HIP backend about 2 months ago
binned_scatter.py

1.5 kB

Add ROCm build artifacts and HIP backend about 2 months ago
cumsum.py

1.5 kB

Add ROCm build artifacts and HIP backend about 2 months ago
gather.py

946 Bytes

Add ROCm build artifacts and HIP backend about 2 months ago
histogram.py

744 Bytes

Add ROCm build artifacts and HIP backend about 2 months ago
histogram_benchmark.py

2.15 kB

Add ROCm build artifacts and HIP backend about 2 months ago
matmul_benchmark.py

12.5 kB

Add ROCm build artifacts and HIP backend about 2 months ago
padded_gather.py

1.26 kB

Add ROCm build artifacts and HIP backend about 2 months ago
padded_scatter.py

2.3 kB

Add ROCm build artifacts and HIP backend about 2 months ago
padded_scatter_benchmark.py

1.89 kB

Add ROCm build artifacts and HIP backend about 2 months ago
permute_benchmark.py

4.94 kB

Add ROCm build artifacts and HIP backend about 2 months ago
repeat.py

215 Bytes

Add ROCm build artifacts and HIP backend about 2 months ago
replicate.py

1.15 kB

Add ROCm build artifacts and HIP backend about 2 months ago
round_up.py

418 Bytes

Add ROCm build artifacts and HIP backend about 2 months ago
scatter.py

1.89 kB

Add ROCm build artifacts and HIP backend about 2 months ago
sort.py

1.07 kB

Add ROCm build artifacts and HIP backend about 2 months ago
sort_benchmark.py

2.21 kB

Add ROCm build artifacts and HIP backend about 2 months ago
stk_autocast.py

1.27 kB

Add ROCm build artifacts and HIP backend about 2 months ago
sum.py

207 Bytes

Add ROCm build artifacts and HIP backend about 2 months ago
topology.py

1.15 kB

Add ROCm build artifacts and HIP backend about 2 months ago