求助贴 优化

综合讨论区
#1

这部分代码执行很慢，能优化吗？
看起来像是一个一个地往GPU里面写，能一次性写进去吗？

@ti.dataclass
class MyObject:
    aa: float
    bb: vec3

    @ti.func
    def get_value(self):
        return self.aa * self.bb


N = 100000
objects = MyObject.field()
ti.root.dense(ti.i, N).place(objects)
for i in range(N):
    objects[i] = MyObject(i / 10, vec3(i // 100))
#2

vec3的定义是什么？

#3

from taichi.math import vec3

#4

你的代码这里

for i in range(N):
    objects[i] = MyObject(i / 10, vec3(i // 100))

感觉应该在taichi kernel并行写入啊，不应该放在python scope串行执行吧