1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
// (c) Meta Platforms, Inc. and affiliates. Confidential and proprietary.
#pragma once
#include <cuda.h>
namespace kineto {
// Warms up CUDA before the tracing starts
void warmup(void);
// Basic usage of cudaMemcpy and cudaMemset
void basicMemcpyToDevice(void);
// Basic usage of cudaMemcpy from device to host
void basicMemcpyFromDevice(void);
// Your experimental code goes in here!
void playground(void);
// Run a simple elementwise kernel
void compute(void);
}
|
