没有合适的资源?快使用搜索试试~ 我知道了~
tensor_compute:使用WebGPU探索GPU计算
共99个文件
rs:46个
comp:21个
spv:21个
需积分: 9 1 下载量 83 浏览量
2021-05-30
10:57:05
上传
评论
收藏 2.1MB ZIP 举报
温馨提示
Rust 张量计算 这是一个自学WebGPU计算的个人项目,重点关于机器学习应用程序。 目前的特点: 选择要使用的 GPU(如果系统中超过 1 个) 克隆 (批量) Matmul 雷卢 转置 充满 比较 使连续 [-] 切片(需要更多设计/工作) [-] 索引(需要更多设计/工作) 创建视图张量 进行中: Element Wise Tensor - Tensor Ops: 添加 子 穆尔 标量 - 张量操作: 添加 子 穆尔 工作示例: fn main () { println! ( "Running in {:?}" , GpuStore :: get_default (). info ()); let ma = Tensor :: from_data_and_shape ( vec! [ 1 ., 2 ., 3 ., 4 ., 5
资源推荐
资源详情
资源评论
收起资源包目录
tensor_compute-master.zip (99个子文件)
tensor_compute-master
compile_shaders.sh 204B
.gitignore 52B
README.md 1KB
Cargo.toml 518B
dev_log.md 200B
examples
simple_mm
main.rs 592B
assignment
main.rs 470B
src
gpu_store
mod.rs 2KB
autograd
ops.rs 1KB
autograd.rs 6KB
tensors
cpu_tensor.rs 7KB
mod.rs 18KB
prelude.rs 67B
gpu_tensor
accessors_contructors.rs 4KB
mod.rs 2KB
gpu_ops
fill_with
fill_with.spv 900B
fill_with.comp 356B
mod.rs 755B
tests.rs 414B
compare
compare.spv 4KB
mod.rs 1KB
compare.comp 971B
tests.rs 497B
bmm
bmm.spv 2KB
mod.rs 3KB
bmm.comp 1KB
tests.rs 3KB
log_soft_max
log_soft_max.spv 740B
mod.rs 1KB
tests.rs 445B
log_soft_max.comp 443B
relu
relu.spv 1004B
mod.rs 1KB
tests.rs 421B
relu.comp 411B
mod.rs 1KB
shared_shader_fragments
el_number_to_lin_mem_idx.comph 2KB
tensor_b_definition.comph 575B
tensor_a_definition.comph 575B
el_number_to_lin_mem_idx_new.comph 2KB
assign
assign.spv 3KB
mod.rs 956B
tests.rs 541B
assign.comp 435B
make_contiguous
make_contiguous.comp 465B
mod.rs 1KB
tests.rs 452B
make_contiguous.spv 3KB
transpose
transpose.comp 1KB
mod.rs 911B
transpose.spv 2KB
tests.rs 603B
binary_ops
pow_scalar.comp 466B
sub.spv 1008B
add.spv 1008B
div_scalar.comp 462B
sub_scalar.comp 462B
add_scalar.comp 462B
mod.rs 3KB
sub_scalar.spv 1KB
dot_div.spv 1008B
dot_mul.comp 608B
div_scalar.spv 1KB
sub.comp 608B
pow_scalar.spv 1KB
add.comp 608B
add_scalar.spv 1KB
dot_mul.spv 1008B
tests.rs 3KB
mul_scalar.comp 462B
mul_scalar.spv 1KB
dot_div.comp 608B
unary_ops
sum.comp 581B
exp.comp 440B
ln.comp 440B
clone.comp 435B
sum.rs 2KB
mod.rs 3KB
clone.spv 796B
sum.spv 1KB
exp.spv 820B
ln.spv 820B
traits.rs 2KB
utils.rs 1KB
indexing
index_to_dim_stride.rs 5KB
mod.rs 4KB
tests.rs 745B
shape_changing
broadcast_tests.rs 2KB
mod.rs 669B
slicing_tests.rs 326B
traits.rs 2KB
gpu_internals
gpu_factory.rs 2KB
mod.rs 531B
shader_runner
shader_runner.rs 5KB
mod.rs 345B
gpu_buffers
mod.rs 5KB
lib.rs 487B
Cargo.lock 46KB
glslc 7.93MB
共 99 条
- 1
资源评论
PaytonSun
- 粉丝: 20
- 资源: 4577
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
最新资源
- 本科毕业设计基于C# wpf人脸识别的考勤系统的设计与实现源码.zip
- 基于Ruoyi+uniapp实现学生考勤系统 学生考勤源码+项目说明.zip
- feae6bc968ca68a099455d8b8a8dea35
- 基于Pytorch训练CIRAR10上分类算法.zip
- Pytorch-pytorch深度学习教程之Tensorboard.zip
- 基于C++和Python开发yolov8-face作为人脸检测器dlib作为人脸识别器的人脸考勤系统源码+项目说明.zip
- Pytorch-pytorch深度学习教程之变分自动编码器.zip
- Pytorch-pytorch深度学习教程之神经风格迁移.zip
- Pytorch-pytorch深度学习教程之深度残差网络.zip
- Pytorch-pytorch深度学习教程之循环神经网络.zip
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功