|
|
之前我们介绍太极语言的ti.kernel内核的时候,提到它支持的参数类型,其中举例所i32和f32这两种,这是太极编程语言中的数据类型。机智客觉得在编程语言中,数据类型广泛用于函数和方法的参数中,比如在kernel中,比如在ti.field方法中,所以是基础,重要性毋庸置疑。3 f3 C3 |; A# w* } t1 J
* P4 s1 N: e- d5 A6 M
和其他编程语言类似,太极目前的数据类型也有几种:有符号整数类型,无符号整数类型,浮点类型。太极不同的数据都是由数据类型和精度组成的。其中,Signed integers包含ti.i8,ti.i16,ti.i32,ti.i64。Unsigned integers类型则用u为前缀,比如ti.u8,ti.u16等。而浮点则有两种,ti.f32和ti.f64。机智客在以前的文章里提到的i32和f32,其实是太极语言里用得最普遍的。当然和其他编程语言不太一样的也有,太极语言的数据类型并不多,布尔类型则是由ti.i32表示。
4 f! ]; Y! P# C, L6 t+ ]9 C7 W
默认的数据类型精度都是32位。需要说明的是,CPU和CUDA后端支持所有的数据类型,而其他则由于后端API的约束,要参考相应后端的API才能确定是否缺少某些数据类型支持。
6 E# h! `) m8 v9 ^( c' O" V. F6 e; k; R. B
上面是太极语言的数据类型,可用于函数里的参数。太极语言规定只能是标量,返回也是。如果函数有返回值,则要指定返回数据类型。而且返回值也只能是一个标量(不能是两个以上),而不能是矢量或矩阵,哪怕Python中的元组也不行。所以参数可以有多个标量,而返回只能是一个标量。比如这段来源于英文文档中的代码片段。
" z' _; R" }7 P4 U; R" _
/ Q6 k8 m( O, y ti.kernel
/ b8 J/ Z8 N+ g- J% M" J4 M$ j. {) @* E% W) K* E) j
def my_kernel()->ti.f32:
1 g6 o8 m- P' a* m2 P) }& C e$ t3 X" S) ]4 B I$ m+ f
return 233.33
! \7 N* k! ~: e" `5 l" Q* x( R
5 S$ z3 [, ^& k6 ? |) `. p 我们以前介绍过,太极语言的初始化,说过,初始化,也可以指定数据类型。这里说下,如果我们要设置初始化的数据类型,可以这样ti.init(default_ip=ti.i64,default_fp=ti.f32)。当然了,数据类型这块,还有很多其他细节知识,比如数据类型转化,矢量和矩阵的知识等。我们以后遇到再学习。
: B% k& g9 p: a, ^/ S- G0 ~ O3 c3 t l ?6 T
taichi而关于太极语言的作用域,相信学习了上文或者以前文章的朋友,都了解了。其实就是 ti.kernel或 ti.func装饰的代码区域内。ti.kernel里的是太极内核,ti.func里的是太极函数。可以调用ti.func里的函数,也可以嵌套调用函数,然而不支持递归函数。外部就是普通Python代码了,不同代码不同处理,各找各妈,各归各管。如果用CUDA做类比的话,Taichi作用域等于device side。taichi https://taichi-lang.cn/
; |9 o* m3 H. @! e9 x6 y% Q3 d8 O' O j$ E, N
7 [# u8 e8 d/ w |
|