Home
last modified time | relevance | path

Searched refs:max_shared_memory_per_block (Results 1 – 25 of 27) sorted by relevance

12

/dports/misc/tvm/incubator-tvm-0.6.1/tests/python/unittest/
H A Dtest_pass_verify_gpu_code.py53 max_shared_memory_per_block=type_size * M - 1,
60 max_shared_memory_per_block=type_size * M,
125 max_shared_memory_per_block=0,
132 max_shared_memory_per_block=0,
139 max_shared_memory_per_block=0,
147 max_shared_memory_per_block=0,
175 max_shared_memory_per_block=0,
182 max_shared_memory_per_block=0,
/dports/misc/py-tvm/incubator-tvm-0.6.1/tests/python/unittest/
H A Dtest_pass_verify_gpu_code.py53 max_shared_memory_per_block=type_size * M - 1,
60 max_shared_memory_per_block=type_size * M,
125 max_shared_memory_per_block=0,
132 max_shared_memory_per_block=0,
139 max_shared_memory_per_block=0,
147 max_shared_memory_per_block=0,
175 max_shared_memory_per_block=0,
182 max_shared_memory_per_block=0,
/dports/misc/tvm/incubator-tvm-0.6.1/src/pass/
H A Dverify_gpu_code.cc38 int64_t max_shared_memory_per_block, in Verify() argument
44 max_shared_memory_per_block_ = static_cast<size_t>(max_shared_memory_per_block); in Verify()
174 int64_t max_shared_memory_per_block = INT64_MAX; in VerifyGPUCode() local
185 max_shared_memory_per_block = val->value; in VerifyGPUCode()
200 max_shared_memory_per_block, in VerifyGPUCode()
/dports/misc/py-tvm/incubator-tvm-0.6.1/src/pass/
H A Dverify_gpu_code.cc38 int64_t max_shared_memory_per_block, in Verify() argument
44 max_shared_memory_per_block_ = static_cast<size_t>(max_shared_memory_per_block); in Verify()
174 int64_t max_shared_memory_per_block = INT64_MAX; in VerifyGPUCode() local
185 max_shared_memory_per_block = val->value; in VerifyGPUCode()
200 max_shared_memory_per_block, in VerifyGPUCode()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/unittest/
H A Dtest_tir_analysis_verify_gpu_code.py64 max_shared_memory_per_block=type_size * M - 1,
81 max_shared_memory_per_block=type_size * M,
176 valid, max_shared_memory_per_block=0, max_threads_per_block=N - 1
191 valid, max_shared_memory_per_block=0, max_threads_per_block=N
207 max_shared_memory_per_block=0,
225 max_shared_memory_per_block=0,
264 valid, max_shared_memory_per_block=0, max_threads_per_block=N - 1
279 valid, max_shared_memory_per_block=0, max_threads_per_block=N
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/tir/analysis/
H A Dverify_gpu_code.cc39 int64_t max_shared_memory_per_block, int64_t max_threads_per_block, in Verify() argument
43 max_shared_memory_per_block_ = static_cast<size_t>(max_shared_memory_per_block); in Verify()
247 int64_t max_shared_memory_per_block = INT64_MAX; in VerifyGPUCode_() local
260 max_shared_memory_per_block = val->value; in VerifyGPUCode_()
278 return verifier.Verify(func->body, max_local_memory_per_block, max_shared_memory_per_block, in VerifyGPUCode_()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/rust/tvm-rt/src/
H A Dcontext.rs35 fn max_shared_memory_per_block(&self) -> isize; in max_shared_memory_per_block() method
80 (max_shared_memory_per_block, 3);
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/include/tvm/auto_scheduler/
H A Dsearch_task.h49 int max_shared_memory_per_block{INT32_MAX};
63 v->Visit("max_shared_memory_per_block", &max_shared_memory_per_block); in VisitAttrs()
/dports/devel/vexcl/vexcl-1.4.3/vexcl/backend/cuda/
H A Dkernel.hpp176 size_t max_shared_memory_per_block(const command_queue &q) const { in max_shared_memory_per_block() function in vex::backend::cuda::kernel
177 return q.device().max_shared_memory_per_block() - in max_shared_memory_per_block()
187 size_t max_smem = max_shared_memory_per_block(q); in config()
H A Dcontext.hpp142 size_t max_shared_memory_per_block() const { in max_shared_memory_per_block() function in vex::backend::cuda::device
/dports/devel/vexcl/vexcl-1.4.3/vexcl/backend/compute/
H A Dkernel.hpp168 size_t max_shared_memory_per_block(const boost::compute::command_queue &q) const { in max_shared_memory_per_block() function in vex::backend::compute::kernel
187 size_t max_smem = max_shared_memory_per_block(queue); in config()
/dports/devel/vexcl/vexcl-1.4.3/vexcl/backend/opencl/
H A Dkernel.hpp180 size_t max_shared_memory_per_block(const cl::CommandQueue &q) const { in max_shared_memory_per_block() function in vex::backend::opencl::kernel
200 size_t max_smem = max_shared_memory_per_block(queue); in config()
/dports/devel/vexcl/vexcl-1.4.3/vexcl/backend/jit/
H A Dcontext.hpp53 size_t max_shared_memory_per_block() const { return 32768UL; } in max_shared_memory_per_block() function
H A Dkernel.hpp174 size_t max_shared_memory_per_block(const command_queue&) const { in max_shared_memory_per_block() function in vex::backend::jit::kernel
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/auto_scheduler/
H A Dsearch_task.cc61 p_hardware_params->max_shared_memory_per_block = ret; in GetDefaultHardwareParams()
/dports/misc/tvm/incubator-tvm-0.6.1/rust/frontend/src/
H A Dcontext.rs266 (max_shared_memory_per_block, 3);
/dports/misc/py-tvm/incubator-tvm-0.6.1/rust/frontend/src/
H A Dcontext.rs266 (max_shared_memory_per_block, 3);
/dports/misc/tvm/incubator-tvm-0.6.1/python/tvm/_ffi/
H A Druntime_ctypes.py192 def max_shared_memory_per_block(self): member in TVMContext
/dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/_ffi/
H A Druntime_ctypes.py192 def max_shared_memory_per_block(self): member in TVMContext
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/_ffi/
H A Druntime_ctypes.py222 def max_shared_memory_per_block(self): member in TVMContext
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/autotvm/measure/
H A Dmeasure_methods.py283 "max_shared_memory_per_block": ctx.max_shared_memory_per_block,
/dports/misc/tvm/incubator-tvm-0.6.1/python/tvm/autotvm/measure/
H A Dmeasure_methods.py236 'max_shared_memory_per_block': ctx.max_shared_memory_per_block,
/dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/autotvm/measure/
H A Dmeasure_methods.py236 'max_shared_memory_per_block': ctx.max_shared_memory_per_block,
/dports/devel/vexcl/vexcl-1.4.3/vexcl/fft/
H A Dkernels.hpp277 const auto local_size = queue.device().max_shared_memory_per_block(); in transpose_kernel()
/dports/devel/vexcl/vexcl-1.4.3/vexcl/
H A Dstencil.hpp414 size_t max_smem = fast_krn.max_shared_memory_per_block(queue[d]); in init()

12