Ailing Zhang ailzhang

ailzhang / example.json

Created November 8, 2021 01:16


	{
	"aot_data": {
	"kernels": {
	"0": {
	"program": {
	"kernels": {
	"0": {
	"kernel_name" : "init_c40_00",
	"kernel_src" : ".//taichi_example.tcb_init_c40_00.glsl",

ailzhang / gist:d669fd809709395a342835ae4077603b

Created November 11, 2021 02:21

	// Test Code

	import taichi as ti
	ti.init(arch=ti.vulkan,log_level=ti.TRACE)

	def test_snode_read_write():
	dtype = ti.f16
	x = ti.field(dtype, shape=())
	x[None] = 0.3
	print(x[None])

ailzhang / collect.py

Created December 13, 2021 01:35

	import taichi as ti
	from types import ModuleType, FunctionType
	import inspect
	import pdb
	import csv

	# FIXME: Without this init some calls will fail
	ti.init()

	def has_doc_string(x):

ailzhang / mpm88_ndarray.py

Created February 10, 2022 06:06

	import os, json
	import taichi as ti
	ti.init(arch=ti.opengl, allow_nv_shader_extension=False, use_gles=True, ndarray_use_torch=False)
	dim = 2
	N = 64
	n_particles = N * N * 2
	n_grid = 128
	p_rho = 1
	bound = 3
	E = 400

ailzhang / repro.py

Created March 4, 2022 11:56

	import paddle
	Python 3.8.11 (default, Aug 3 2021, 15:09:35)
	Type 'copyright', 'credits' or 'license' for more information
	IPython 7.26.0 -- An enhanced Interactive Python. Type '?' for help.

	In [1]: import paddle

	In [2]: import numpy as np
	In [4]: x = paddle.to_tensor(2 * np.ones((4, ), dtype=np.flo
	...: at32))

ailzhang / demo.py

Created July 28, 2022 04:54

	import taichi as ti
	ti.init()

	# you must make sure the func body here works both in python and taichi scope.
	def f():
	a = 0
	for i in range(10):
	a += i
	print(a)

ailzhang / ndarray_playground.py

Created November 10, 2022 04:56