Skip to content

Commit

Permalink
Fixed formatting issues
Browse files Browse the repository at this point in the history
  • Loading branch information
DoeringChristian committed Nov 29, 2024
1 parent c7fc684 commit ef227f5
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 10 deletions.
9 changes: 3 additions & 6 deletions src/record_ts.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -769,11 +769,8 @@ int Recording::replay_launch(Operation &op) {
}

// Change kernel size in `kernel_params`
if (backend == JitBackend::CUDA) {
uintptr_t size = 0;
std::memcpy(&size, &launch_size, sizeof(uint32_t));
kernel_params[0] = (void *) size;
}
if (backend == JitBackend::CUDA)
kernel_params[0] = (void *) (uintptr_t) launch_size;

if (!dry_run) {
#ifndef NDEBUG
Expand Down Expand Up @@ -1530,7 +1527,7 @@ void RecordThreadState::record_aggregate(void *dst, AggregationEntry *agg,
size);

uint32_t start = m_recording.dependencies.size();

add_out_param(dst, VarType::UInt8);

for (uint32_t i = 0; i < size; ++i) {
Expand Down
6 changes: 3 additions & 3 deletions src/record_ts.h
Original file line number Diff line number Diff line change
Expand Up @@ -195,7 +195,7 @@ struct Operation {
struct {
/// The reduce type of a prefix reduction operation
ReduceOp rtype;

/// Whether a prefix sum operation is exclusive
bool exclusive;
bool reverse;
Expand Down Expand Up @@ -450,7 +450,7 @@ struct RecordThreadState : ThreadState {
protected:
/// Indicates that recording has been paused.
bool m_paused = false;

/// Mapping from data pointer of a variable to a index into the slot of the
/// recording.
PtrToSlot ptr_to_slot;
Expand Down Expand Up @@ -556,7 +556,7 @@ struct RecordThreadState : ThreadState {
* function will create a new variable and mapping from the pointer to it.
*/
void notify_free(const void *ptr) override;

// =============================================================

~RecordThreadState() {}
Expand Down
2 changes: 1 addition & 1 deletion tests/test.h
Original file line number Diff line number Diff line change
Expand Up @@ -337,7 +337,7 @@ template <typename Func> class FrozenFunction {
return record(input_vector, args...);
} else {
std::vector<uint32_t> output_vector(m_outputs, 0);

jit_log(LogLevel::Debug, "replay:");
// replay adds borrowing references to the \c output_vector
jit_freeze_replay(m_recording, input_vector.data(),
Expand Down

0 comments on commit ef227f5

Please sign in to comment.