mirror of https://github.com/commaai/tinygrad.git
strip whitespace
This commit is contained in:
parent
9eb2746d62
commit
c42d2c4731
|
@ -1,6 +1,12 @@
|
|||
repos:
|
||||
- repo: local
|
||||
hooks:
|
||||
- id: whitespace
|
||||
name: strip whitespace
|
||||
entry: ./strip_whitespace.sh
|
||||
language: system
|
||||
always_run: true
|
||||
pass_filenames: false
|
||||
- id: mypy
|
||||
name: mypy
|
||||
entry: mypy tinygrad/ extra/helpers.py
|
||||
|
|
|
@ -6,7 +6,7 @@ class CUDALanguage(CStyleLanguage):
|
|||
smem_prefix = "__shared__ "
|
||||
smem_prefix_for_cast = False
|
||||
arg_int_prefix = "const int"
|
||||
barrier = "__syncthreads();"
|
||||
barrier = "__syncthreads();"
|
||||
float4 = "make_float4"
|
||||
gid = [f'blockIdx.{chr(120+i)}' for i in range(3)]
|
||||
lid = [f'threadIdx.{chr(120+i)}' for i in range(3)]
|
||||
|
|
|
@ -19,7 +19,7 @@ class HIPLanguage(CStyleLanguage):
|
|||
smem_prefix_for_cast=False
|
||||
barrier = "__syncthreads();"
|
||||
float4 = "make_float4"
|
||||
uses_vload=True
|
||||
uses_vload=True
|
||||
uses_ptr_arithmetic=True
|
||||
arg_int_prefix = "const int"
|
||||
half_prekernel = "#include <hip/hip_fp16.h>\nusing half4 = HIP_vector_type<half, 4>;" + """
|
||||
|
|
Loading…
Reference in New Issue