/dports/misc/mxnet/incubator-mxnet-1.9.0/cpp-package/example/ |
H A D | mlp.cpp | 75 Context ctx_dev(DeviceType::kCPU, 0); in MLP() local 77 NDArray array_x(Shape(128, 28), ctx_dev, false); in MLP() 78 NDArray array_y(Shape(128), ctx_dev, false); in MLP() 96 NDArray array_w_1(Shape(512, 28), ctx_dev, false); in MLP() 97 NDArray array_b_1(Shape(512), ctx_dev, false); in MLP() 98 NDArray array_w_2(Shape(10, 512), ctx_dev, false); in MLP() 99 NDArray array_b_2(Shape(10), ctx_dev, false); in MLP() 110 NDArray array_w_1_g(Shape(512, 28), ctx_dev, false); in MLP() 111 NDArray array_b_1_g(Shape(512), ctx_dev, false); in MLP() 112 NDArray array_w_2_g(Shape(10, 512), ctx_dev, false); in MLP() [all …]
|
H A D | lenet.cpp | 37 ctx_dev(Context(DeviceType::kCPU, 0)) in Lenet() 39 ctx_dev(Context(DeviceType::kGPU, 0)) in Lenet() 122 args_map["data"] = data_array.Slice(0, batch_size).Copy(ctx_dev); in Run() 123 args_map["data_label"] = label_array.Slice(0, batch_size).Copy(ctx_dev); in Run() 138 lenet.InferArgsMap(ctx_dev, &args_map, args_map); in Run() 146 Executor *exe = lenet.SimpleBind(ctx_dev, args_map); in Run() 157 .Copy(ctx_dev); in Run() 160 .Copy(ctx_dev); in Run() 182 Context ctx_dev; member in Lenet 221 val_data.Slice(start_index, start_index + batch_size).Copy(ctx_dev); in ValAccuracy() [all …]
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/cpp-package/example/ |
H A D | mlp.cpp | 75 Context ctx_dev(DeviceType::kCPU, 0); in MLP() local 77 NDArray array_x(Shape(128, 28), ctx_dev, false); in MLP() 78 NDArray array_y(Shape(128), ctx_dev, false); in MLP() 96 NDArray array_w_1(Shape(512, 28), ctx_dev, false); in MLP() 97 NDArray array_b_1(Shape(512), ctx_dev, false); in MLP() 98 NDArray array_w_2(Shape(10, 512), ctx_dev, false); in MLP() 99 NDArray array_b_2(Shape(10), ctx_dev, false); in MLP() 110 NDArray array_w_1_g(Shape(512, 28), ctx_dev, false); in MLP() 111 NDArray array_b_1_g(Shape(512), ctx_dev, false); in MLP() 112 NDArray array_w_2_g(Shape(10, 512), ctx_dev, false); in MLP() [all …]
|
H A D | lenet.cpp | 37 ctx_dev(Context(DeviceType::kCPU, 0)) in Lenet() 39 ctx_dev(Context(DeviceType::kGPU, 0)) in Lenet() 122 args_map["data"] = data_array.Slice(0, batch_size).Copy(ctx_dev); in Run() 123 args_map["data_label"] = label_array.Slice(0, batch_size).Copy(ctx_dev); in Run() 138 lenet.InferArgsMap(ctx_dev, &args_map, args_map); in Run() 146 Executor *exe = lenet.SimpleBind(ctx_dev, args_map); in Run() 157 .Copy(ctx_dev); in Run() 160 .Copy(ctx_dev); in Run() 182 Context ctx_dev; member in Lenet 221 val_data.Slice(start_index, start_index + batch_size).Copy(ctx_dev); in ValAccuracy() [all …]
|
/dports/math/onednn/oneDNN-2.5.1/src/sycl/ |
H A D | sycl_engine.cpp | 65 [&](const ::sycl::device &ctx_dev) { in engine_create() argument 66 return are_equal(ctx_dev, dev); in engine_create()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/sycl/ |
H A D | sycl_engine.cpp | 65 [&](const cl::sycl::device &ctx_dev) { in engine_create() argument 66 return are_equal(ctx_dev, dev); in engine_create()
|
/dports/lang/gcc6-aux/gcc-6-20180516/libgomp/plugin/ |
H A D | plugin-nvptx.c | 572 CUdevice dev, ctx_dev; in nvptx_open_device() local 586 r = cuCtxGetDevice (&ctx_dev); in nvptx_open_device() 590 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
|
/dports/lang/gnat_util/gcc-6-20180516/libgomp/plugin/ |
H A D | plugin-nvptx.c | 572 CUdevice dev, ctx_dev; in nvptx_open_device() local 586 r = cuCtxGetDevice (&ctx_dev); in nvptx_open_device() 590 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
|
/dports/lang/gcc10/gcc-10.3.0/libgomp/plugin/ |
H A D | plugin-nvptx.c | 402 CUdevice dev, ctx_dev; in nvptx_open_device() local 414 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device() 421 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
|
/dports/devel/avr-gcc/gcc-10.2.0/libgomp/plugin/ |
H A D | plugin-nvptx.c | 402 CUdevice dev, ctx_dev; in nvptx_open_device() local 414 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device() 421 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
|
/dports/lang/gcc10-devel/gcc-10-20211008/libgomp/plugin/ |
H A D | plugin-nvptx.c | 402 CUdevice dev, ctx_dev; in nvptx_open_device() local 414 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device() 421 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
|
/dports/lang/gcc12-devel/gcc-12-20211205/libgomp/plugin/ |
H A D | plugin-nvptx.c | 420 CUdevice dev, ctx_dev; in nvptx_open_device() local 432 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device() 439 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
|
/dports/lang/gcc8/gcc-8.5.0/libgomp/plugin/ |
H A D | plugin-nvptx.c | 714 CUdevice dev, ctx_dev; in nvptx_open_device() local 726 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device() 733 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
|
/dports/devel/riscv64-none-elf-gcc/gcc-8.4.0/libgomp/plugin/ |
H A D | plugin-nvptx.c | 714 CUdevice dev, ctx_dev; in nvptx_open_device() local 726 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device() 733 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
|
/dports/lang/gcc11-devel/gcc-11-20211009/libgomp/plugin/ |
H A D | plugin-nvptx.c | 419 CUdevice dev, ctx_dev; in nvptx_open_device() local 431 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device() 438 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
|
/dports/devel/arm-none-eabi-gcc/gcc-8.4.0/libgomp/plugin/ |
H A D | plugin-nvptx.c | 714 CUdevice dev, ctx_dev; in nvptx_open_device() local 726 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device() 733 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
|
/dports/lang/gcc9-devel/gcc-9-20211007/libgomp/plugin/ |
H A D | plugin-nvptx.c | 741 CUdevice dev, ctx_dev; in nvptx_open_device() local 753 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device() 760 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
|
/dports/devel/riscv32-unknown-elf-gcc/gcc-8.4.0/libgomp/plugin/ |
H A D | plugin-nvptx.c | 714 CUdevice dev, ctx_dev; in nvptx_open_device() local 726 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device() 733 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
|
/dports/devel/aarch64-none-elf-gcc/gcc-8.4.0/libgomp/plugin/ |
H A D | plugin-nvptx.c | 714 CUdevice dev, ctx_dev; in nvptx_open_device() local 726 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device() 733 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
|
/dports/misc/cxx_atomics_pic/gcc-11.2.0/libgomp/plugin/ |
H A D | plugin-nvptx.c | 419 CUdevice dev, ctx_dev; in nvptx_open_device() local 431 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device() 438 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
|
/dports/devel/riscv64-gcc/gcc-8.3.0/libgomp/plugin/ |
H A D | plugin-nvptx.c | 714 CUdevice dev, ctx_dev; in nvptx_open_device() local 726 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device() 733 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
|
/dports/lang/gcc11/gcc-11.2.0/libgomp/plugin/ |
H A D | plugin-nvptx.c | 419 CUdevice dev, ctx_dev; in nvptx_open_device() local 431 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device() 438 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
|
/dports/lang/gcc9/gcc-9.4.0/libgomp/plugin/ |
H A D | plugin-nvptx.c | 741 CUdevice dev, ctx_dev; in nvptx_open_device() local 753 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device() 760 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
|
/dports/lang/gcc9-aux/gcc-9.1.0/libgomp/plugin/ |
H A D | plugin-nvptx.c | 741 CUdevice dev, ctx_dev; in nvptx_open_device() local 753 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device() 760 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
|