supplient

The MIT License (MIT)

Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:

	import torch
	import taichi as ti
	ti.init(arch=ti.cuda)

	@ti.func
	def test_dev(x: ti.types.vector(2, ti.f32)):
	return 2*x

	@ti.kernel
	def test_kernel(x: ti.types.ndarray()):

	import numpy as np
	import torch
	import numba.cuda as cu

	@cu.jit
	def enter(upside, down_value, down_index, dumb_R):
	di = cu.grid(1)
	if di >= down_value.shape[0]:
	return
	ui = di * 2

	import torch
	import numba.cuda as cu


	# Convenient function for specifying CUDA thread block size
	_block_size = 512
	def _cal_block_num(n):
	return int((n-1)/_block_size) + 1

	class IncFunction(torch.autograd.Function):

	import numpy as np
	import torch
	import numba.cuda as cu

	@cu.jit
	def enter(upside, downside):
	di = cu.grid(1)
	if di >= downside.shape[0]:
	return
	ai = di * 2

	_path2modname = {}

	def Filepath2ModuleName(filepath):
	''' Get an imported module's qualified name by its file path.

	Return `None` if the module is not found or is not imported.

	## Usage
	``` python
	from pkg import mod

	'''
	# SGR: Select Graphic Rendition

	This module is a cheetsheet for ANSI escape characters which can control terminal text's appearence.

	## usage
	``` python
	from sgr import sgr
	print(f"{sgr.red + sgr.bg_yellow + sgr.bold}test{sgr.reset} test")
	```

	# imports only for type hinting
	from __future__ import annotations
	from typing import TYPE_CHECKING
	if TYPE_CHECKING:
	from importlib.resources import Package
	from types import ModuleType

	def ImportAllModules(pkg: Package)-> list[ModuleType]:
	'''Import all modules in `pkg`.

	import taichi as ti
	import numpy as np
	from random import random
	from tqdm import tqdm
	from math import sin, pi, sqrt
	from timer import Timer

	timer = Timer()

	################

	#include <Windows.h>

	// cuda_helpers.h is my own repo
	// refer to https://github.com/supplient/CudaHelper
	#define CUDA_HELPER_NAMESPACE cuh
	#include <cuda_helpers.h>

	#include <thrust/scan.h>
	#include <thrust/execution_policy.h>
	#include <thrust/fill.h>